Update main.py

main.py

@@ -20,22 +20,21 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-# ✅ Corrected path to relative static directory
 app.mount("/static", StaticFiles(directory="static"), name="static")
 
 @app.get("/")
 def read_root():
     return FileResponse("static/index.html")
 
-# Load pretrained model
-model = models.resnet50(weights=models.ResNet50_Weights.DEFAULT)
+# ✅ Load model without downloading
+model = models.resnet50()
+model.load_state_dict(torch.load("resnet50_weights.pth", map_location="cpu"))
 model.eval()
 
 # Load labels
 with open("imagenet_classes.txt") as f:
     labels = [line.strip() for line in f]
 
-# Define image transforms
 transform = transforms.Compose([
     transforms.Resize(256),
     transforms.CenterCrop(224),