Update app.py

app.py

@@ -4,6 +4,7 @@ from transformers import AutoModelForImageClassification
 import gradio as gr
 import torch
 
+num_classes = 2
 def predict(inp):
     inputs = data_transforms(inp)[None]
     model.eval()