Update app.py

app.py

@@ -1,8 +1,6 @@
-# Load model directly
-from transformers import AutoImageProcessor, AutoModelForImageClassification
+import torch
 
-processor = AutoImageProcessor.from_pretrained("microsoft/resnet-18")
-model = AutoModelForImageClassification.from_pretrained("microsoft/resnet-18")
+model = torch.hub.load('pytorch/vision:v0.6.0', 'resnet18', pretrained=True).eval() # evaluation
 
 import requests # 用於發送 HTTP 請求
 from PIL import Image