Update app.py

app.py

@@ -39,11 +39,7 @@ def transcribe(audio_file):
 # Text Generation Function
 def generate_text(prompt):
     formatted_prompt = format_prompt(prompt)
-    messages = [{"role": "user", "content": formatted_prompt}]
-    response = ""
-    for message in client.chat_completion(messages, max_tokens=512, stream=True, temperature=0.7, top_p=0.95):
-        token = message.choices[0].delta.content
-        response += token
+    response = client.text_generation(formatted_prompt, max_new_tokens=512, temperature=0.7, top_p=0.95)
     return response.strip()
 
 # TTS Function