Update app.py

app.py

@@ -43,7 +43,7 @@ if user_input:
             output = query(checkpoint,{
                 "inputs": prompt,
                 "temperature":temperature,
-                "wait_for_model":True})[0]['generated_text']
+                "wait_for_model":True})
         else:
 
             prompt = user_input
@@ -51,7 +51,7 @@ if user_input:
             output = query(checkpoint,{
                 "inputs": prompt,
                 "temperature":temperature,
-                "wait_for_model":True})[0]['generated_text']
+                "wait_for_model":True})
         
         st.write(f'Model {model_name}: {output}')
 
@@ -74,3 +74,4 @@ if user_input:
         
 #if __name__ == "__main__":
 #    main()
+#[0]['generated_text']