Update app.py

app.py

@@ -37,9 +37,10 @@ def prompt_model(model_id, system_prompt, user_prompt):
       {"role": "assistant", "content": ""}
     ]
 
-    outputs = pipe(messages)
+    output = pipe(messages)
+    print(output)
     
-    return outputs[0]["generated_text"][-1]
+    return output[0]["generated_text"][-1]["content"]
 
 def fine_tune_model(model_id):
     tokenizer = download_model(model_id)