Update app/main.py

app/main.py

@@ -26,7 +26,7 @@ t = EnThTranslator()
 e = ThEnTranslator()
 
 def extract_restext(response, is_chat=False):
-  return response['choices'][0]['text' if is_chat else 'message'].strip()
+  return response['choices'][0]['message' if is_chat else 'text'].strip()
 
 def ask_llama(llm: Llama, question: str, max_new_tokens=200, temperature=0.5, repeat_penalty=2.0):
   prompt = f"""<|begin_of_text|>