updating predict function

app.py

@@ -18,7 +18,7 @@ def predict(text):
     outputs = model(**inputs)
     predictions = outputs.logits
 
-    return mapping[predictions.argmax()]
+    return mapping[predictions.argmax().item()]
 
 iface = gr.Interface(
     fn=predict,