Update app.py

app.py

@@ -23,7 +23,7 @@ bench = Benchmark(model, tokenizer)
 
 def get_prediction(text):
     encoding = tokenizer(text, return_tensors="pt", padding="max_length", truncation=True, max_length=200)
-    encoding = {k: v.to(new_model.device) for k, v in encoding.items()}
+    encoding = {k: v.to(model.device) for k, v in encoding.items()}
 
     with torch.no_grad():
         outputs = model(**encoding)