Update app.py

app.py

@@ -4,16 +4,17 @@ os.system("pip install ollama")
 
 from ollama import chat
 
-stream = chat(
-    model='llama3.2',
-    messages=[{'role': 'user', 'content': 'Why is the sky blue?'}],
-    stream=True,
-)
-output=""
-for chunk in stream:
-    print(chunk['message']['content'], end='', flush=True)
-    output+=chunk['message']['content']
-    yield output
+def chat_with_ollama(prompt):
+    stream = chat(
+        model='llama3.2',
+        messages=[{'role': 'user', 'content': f'{prompt}'}],
+        stream=True,
+    )
+    output=""
+    for chunk in stream:
+        print(chunk['message']['content'], end='', flush=True)
+        output+=chunk['message']['content']
+        yield output
 
     
 # Create a Gradio interface