Spaces:

Pavankalyan
/

S2I

Runtime error

Pavankalyan commited on Sep 4, 2023

Commit

0cecb50

•

1 Parent(s): e5d627d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -26,7 +26,7 @@ def predict(text):
     with torch.no_grad():
         outputs = model(**inputs)
     predicted_labels = postprocess(outputs)
-    ans = intents_dict[predicted_labels[0]]
     return ans
 from transformers import pipeline

     with torch.no_grad():
         outputs = model(**inputs)
     predicted_labels = postprocess(outputs)
+    ans = intents_dict[str(predicted_labels[0])]
     return ans
 from transformers import pipeline