Update app.py

app.py

@@ -28,5 +28,5 @@ st.title("WaltWhitman-GPT By Ilyas")
 
 text = st.text_area("Enter Prompt")
 if text:
-    infer(text)
+    output = infer(text)
     st.write(output)