Update app.py

app.py

@@ -29,11 +29,13 @@ def generate(prompt, history, temperature=0.9, max_new_tokens=256, top_p=0.95, r
     formatted_prompt = format_prompt(prompt, history)
     response = client.text_generation(formatted_prompt, **generate_kwargs)
 
-    if 'choices' in response:
+    if isinstance(response, list) and len(response) > 0 and 'choices' in response[0]:
+        return response[0]['choices'][0]['text']
+    elif 'choices' in response:
         return response['choices'][0]['text']
     else:
         return response['text']
-
+        
 def main():
     st.title("Mistral 7B Chat Interface")