lab2 test

app.py

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