Update app.py

app.py

@@ -17,7 +17,7 @@ def predict(image):
 
   # Make prediction
   with torch.no_grad():
-    outputs = student_model(**inputs)
+    outputs = model(**inputs)
     predicted_class_idx = torch.argmax(outputs.logits, dim=1).item()
 
   # Get predicted class label