Update app.py

app.py

@@ -35,6 +35,9 @@ def main():
                 
                 prompt = 'generate a question: ' + user_input
 
+            else:
+                prompt = user_input
+            
             output = query(checkpoint,{
                         "inputs": prompt,
                         "wait_for_model":True})