Update app.py

app.py

@@ -14,10 +14,8 @@ async def root():
 
 # Load model and processor
 model_name = "ivandrian11/fruit-classifier"
-AutoModelForImageClassification.from_pretrained(model_name).save_pretrained("fruit-model")
-AutoImageProcessor.from_pretrained(model_name).save_pretrained("fruit-model")
-processor = AutoImageProcessor.from_pretrained("fruit-model")
-model = AutoModelForImageClassification.from_pretrained("fruit-model")
+processor = AutoImageProcessor.from_pretrained(model_name)
+model = AutoModelForImageClassification.from_pretrained(model_name)
 model.eval()  # set to evaluation mode
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(DEVICE)