Update app.py

app.py

@@ -44,12 +44,12 @@ My favorite part of the book though was when the hobbits met Tom Bombadil, it's
                   data=input_data,
                   headers=headers).json()
 
-            if 0 not in r.keys():
-                st.write(r)
-                continue
-            else:
+            try:
                 r=r[0]
- 
+            except:
+                st.write("Model loading timed out. Please enter the text again.")
+                continue
+                
             label, score = convert_dict[r['label']], r['score']
 
             st.write(f"**Model Name**: {model_name}")