Update app.py

app.py

@@ -34,10 +34,10 @@ if user_input:
         output = query(checkpoint,{
             "inputs": user_input,
             "temperature":temperature,
-            "wait_for_model":True})[0][0]['generated_text']
+            "wait_for_model":True})
 
         model_name = checkpoints.split('/')[1]
-        st.write(f'Model {model_name}: output')
+        st.write(f'Model {model_name}: {output}')
 
 
 #if __name__ == "__main__":