Update app.py

app.py

@@ -1,3 +1,15 @@
+from transformers import pipeline
+
+def main():
+    messages = [
+        {"role": "system", "content": "You are a pirate chatbot who always responds in pirate speak!"},
+        {"role": "user", "content": "Who are you?"},
+    ]
+    chatbot = pipeline("text-generation", model="mistralai/Mistral-7B-Instruct-v0.3")
+    r=chatbot(messages)
+    print(r)
+
+"""
 import os
 import time
 from datetime import timedelta
@@ -35,6 +47,7 @@ def main():
         elapsed_time, total_time = cpu_benchmark(cores)
         result = f"Elapsed Time per Iteration: {elapsed_time:.6f} seconds\nTotal Time: {total_time:.6f} seconds"
         st.success(result)
-
+"""
 if __name__ == "__main__":
-    main()
+    main()
+