Update app.py

app.py

@@ -4,8 +4,9 @@ from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 from pydub import AudioSegment
 from sentence_transformers import SentenceTransformer, util
 import spacy
-spacy.cli.download("en_core_web_sm")
+spacy.cli download("en_core_web_sm")
 import json
+from faster_whisper import WhisperModel
 import ollama
 
 # Audio conversion from MP4 to MP3
@@ -50,8 +51,8 @@ soap_embeddings = {section: embedder.encode(prompt, convert_to_tensor=True) for
 def ollama_query(user_prompt, soap_note):
     combined_prompt = f"User Instructions:\n{user_prompt}\n\nContext:\n{soap_note}"
     try:
-        response = ollama.run("llama2:7b-uncensored", prompt=combined_prompt)
-        return response
+        response = ollama.chat(model="llama2:7b-uncensored", messages=[{"role": "user", "content": combined_prompt}])
+        return response['text']
     except Exception as e:
         return f"Error generating response: {e}"