Update app.py

app.py

@@ -19,4 +19,6 @@ if text:
     output_ids = model.generate(input_ids.to('cuda'))
     response = tokenizer.decode(output_ids[0][input_ids.shape[1]:], skip_special_tokens=True)
     print(response)
-    st.json(response)
+    st.json({
+        "response": response
+    })