catch error

main_sentiment.py

@@ -32,23 +32,27 @@ def read_root():
 @app.post("/predict")
 def predict(text:str):
     "prediction endpoint, classifying tweets"
-    
-    text = preprocess(text)
-
-    # PyTorch-based models
-    encoded_input = tokenizer(text, return_tensors='pt')
-    output = model(**encoded_input)
-    scores = output[0][0].detach().numpy()
-    scores = softmax(scores)
-
-    #Process scores
-    ranking = np.argsort(scores)
-    ranking = ranking[::-1]  
-    predicted_label = config.id2label[ranking[0]]
-    predicted_score = scores[ranking[0]]
-
-
-    return {"text":text,
-            "predicted_label":predicted_label,
-            "confidence_score":predicted_score
-            }
+    try:
+        text = preprocess(text)
+
+        # PyTorch-based models
+        encoded_input = tokenizer(text, return_tensors='pt')
+        output = model(**encoded_input)
+        scores = output[0][0].detach().numpy()
+        scores = softmax(scores)
+
+        #Process scores
+        ranking = np.argsort(scores)
+        ranking = ranking[::-1]  
+        predicted_label = config.id2label[ranking[0]]
+        predicted_score = scores[ranking[0]]
+
+
+        return {"text":text,
+                "predicted_label":predicted_label,
+                "confidence_score":predicted_score
+                }
+    except Exception as e:
+        return {
+            "error": e
+        }