Update app.py

app.py

@@ -52,9 +52,12 @@ def main():
 
             input_data = json.dumps({'inputs':user_input})
             r = requests.get(url,data=input_data,headers=headers)
-            output = r.json()[0]['generated_text']
+            try:
+                output = r.json()[0]['generated_text']
 
-            st.write(f'**{name}**: {output}')
+                st.write(f'**{name}**: {output}')
+            except:
+                st.write(f'**{name}**: There was an error when calling the model. Please resubmit the question.')
 
         model_engine = "gpt-3.5-turbo"
         # Max tokens to produce