Update app.py

app.py

@@ -17,7 +17,7 @@ def classify_image(image):
     # Make prediction
     with torch.no_grad():
         outputs = model(**inputs)
-        logits1 = outputs.logits1
+        logits1 = outputs.logits
     # Retrieve the highest probability class label index
     predicted_class_idx = logits1.argmax(-1).item()
     # Define a manual mapping of label indices to human-readable labels