lab2 test

app.py

@@ -54,7 +54,7 @@ def submit_to_model():
             {"role": "user", "content": prompt},
         ]
     )
-    return response
+    return response.choices[0].message.content
 
 
 # App