Update app.py

app.py

@@ -42,7 +42,13 @@ My favorite part of the book though was when the hobbits met Tom Bombadil, it's
             input_data = json.dumps({'inputs':user_input})
             r = requests.post(URL,
                   data=input_data,
-                  headers=headers).json()[0]
+                  headers=headers).json()
+
+            if 0 not in r:
+                st.write(r)
+                return
+            else:
+                r=r[0]
  
             label, score = convert_dict[r['label']], r['score']