Update app.py

app.py

@@ -35,10 +35,11 @@ def main():
                 
                 prompt = 'generate a question: ' + user_input
 
-            try:
-                output = query(checkpoint,{
+            output = query(checkpoint,{
                         "inputs": prompt,
-                        "wait_for_model":True})[0]['generated_text']
+                        "wait_for_model":True})
+            try:
+                output=output[0]['generated_text']
             except:
                 st.write(output)
                 return