Update app.py

app.py

@@ -15,9 +15,11 @@ async def root():
     return {"message": "API is running"}
 
 # Load model and processor
-MODEL_NAME = "ivandrian11/fruit-classifier"
-processor = AutoImageProcessor.from_pretrained(MODEL_NAME)
-model = AutoModelForImageClassification.from_pretrained(MODEL_NAME)
+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")
 model.eval()  # set to evaluation mode
 DEVICE = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model.to(DEVICE)