Update app.py

app.py

@@ -6,8 +6,8 @@ import time
  
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
  
-processor = ViTImageProcessor.from_pretrained("LCZs_v2",local_files_only=True)
-model = ViTForImageClassification.from_pretrained("LCZs_v2",local_files_only=True).to(device)
+processor = ViTImageProcessor.from_pretrained("ViT_LCZs_v2",local_files_only=True)
+model = ViTForImageClassification.from_pretrained("ViT_LCZs_v2",local_files_only=True).to(device)
  
 def predict(image):
     inputs = processor(images=image, return_tensors="pt").to(device)