Update app.py

app.py

@@ -99,14 +99,18 @@ async def predict_sentiment(input_data: TextAspectInput):
 
         # Process input
         input_ids = process_text(text, aspect)
+        print("Process text: ", input_ids)
 
         # Make prediction
-        with torch.no_grad():
-            logits = model(input_ids)
-            probabilities = torch.softmax(logits, dim=-1)
-            prediction = probabilities.argmax(dim=-1).item()
-            sentiment = SENTIMENT_LABELS[prediction]
-
+        try:
+            with torch.no_grad():
+                logits = model(input_ids)
+                probabilities = torch.softmax(logits, dim=-1)
+                prediction = probabilities.argmax(dim=-1).item()
+                sentiment = SENTIMENT_LABELS[prediction]
+        except Exception as e:
+            print(e)
+            
         return {"sentiment": sentiment, "probabilities": probabilities.squeeze().tolist()}
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))