Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import torch
+# import torch
 
 # from datasets import DatasetDict, Dataset
 
@@ -32,8 +32,8 @@ input_text = st.input_text(input)
 inputs = tokenizer(input_text, return_tensors="pt")
 
 # Perform inference
-with torch.no_grad():
-    outputs = model(**inputs)
+# with torch.no_grad():
+outputs = model(**inputs)
 
 # Get the predicted class
 predicted_class = outputs.logits