Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
 import pickle
 import numpy as np
+import torch
 from torch import nn
 from transformers import AutoModelForSequenceClassification
 from sklearn.pipeline import Pipeline