Update app.py

app.py

@@ -33,8 +33,7 @@ with open('id_to_category.pkl', 'rb') as fp:
     id_to_category = pickle.load(fp)
 
 def text_classifier(text):
-  predictions=end_to_end_model.predict(text)
-  for pred in predictions:
-    return(id_to_category[np.argmax(pred)])
+  pred=end_to_end_model.predict([text])
+  return(id_to_category[np.argmax(pred)])
 
 iface= gr.Interface(fn=text_classifier, inputs="text", outputs="text").launch()