Update app.py

app.py

@@ -38,18 +38,25 @@ processor = ViTImageProcessor(
 
 print("Loading custom model...")
 
-config = AutoConfig.from_pretrained(
-    MODEL_REPO,
-    token=HF_TOKEN,
-    trust_remote_code=True,
-)
+# config = AutoConfig.from_pretrained(
+#     MODEL_REPO,
+#     token=HF_TOKEN,
+#     trust_remote_code=True,
+# )
+
+# model = AutoModelForImageClassification.from_pretrained(
+#     MODEL_REPO,
+#     config=config,
+#     token=HF_TOKEN,
+#     trust_remote_code=True,
+#     output_attentions=True,
+# )
 
 model = AutoModelForImageClassification.from_pretrained(
     MODEL_REPO,
-    config=config,
     token=HF_TOKEN,
     trust_remote_code=True,
-    output_attentions=True,
+    output_attentions=True
 )
 
 model.to(DEVICE)