model update

app.py

@@ -13,4 +13,4 @@ demo = gr.Interface(
     description="Upload an image to get the predicted class with a sample image and top-5 prediction chart."
 )
 
-demo.launch()
+demo.launch(share=True, ssr_mode=False)

predict.py

@@ -36,7 +36,7 @@ class SwinCustom(nn.Module):
         outputs = self.model(images)
         return outputs.logits
 
-model_path = hf_hub_download(repo_id="Noha90/AML_16", filename="large_swin_best_model.pth")
+model_path = hf_hub_download(repo_id="surateakash/scene_classification_swin_large", filename="large_swin_best_model.pth")
 print("Model path:", model_path)
 model = SwinCustom(model_name=MODEL_NAME, num_classes=40)
 state_dict = torch.load(model_path, map_location="cpu")