Upload Inference_Restaurant_Review.py

Inference_Restaurant_Review.py

@@ -25,7 +25,7 @@ def inference(text):
     with torch.no_grad():
         logits = model(**inputs).logits
     predicted_class_id = logits.argmax().item()
-    print(predicted_class_id)
+    print((predicted_class_id+2)/2)
     
 inference(text1)
 inference(text2)