Update app.py

app.py

@@ -1,6 +1,5 @@
 
 
-     
 import torch
 from transformers import CLIPModel, CLIPProcessor, AutoTokenizer, MarianMTModel, MarianTokenizer
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
@@ -105,11 +104,16 @@ class EnhancedBanglaSDGenerator:
 
     def _load_banglaclip_model(self, weights_path: str) -> CLIPModel:
         try:
-            if not Path(weights_path).exists():
-                raise FileNotFoundError(f"BanglaCLIP weights not found at {weights_path}")
+            model_file_path = Path("model_cache/banglaclip_model_epoch_10.pth")
+            if not model_file_path.exists():
+                logger.info("Downloading BanglaCLIP weights...")
+                torch.hub.download_url_to_file(
+                    'https://huggingface.co/Mansuba/Bangla_text_to_image_app/resolve/main/banglaclip_model_epoch_10.pth',
+                    model_file_path
+                )
 
             clip_model = CLIPModel.from_pretrained(self.clip_model_name)
-            state_dict = torch.load(weights_path, map_location=self.device)
+            state_dict = torch.load(model_file_path, map_location=self.device)
 
             cleaned_state_dict = {
                 k.replace('module.', '').replace('clip.', ''): v
@@ -240,7 +244,7 @@ def create_gradio_interface():
         nonlocal generator
         if generator is None:
             generator = EnhancedBanglaSDGenerator(
-                banglaclip_weights_path="banglaclip_model_epoch_10.pth",
+                banglaclip_weights_path="model_cache/banglaclip_model_epoch_10.pth",
                 cache_dir=str(cache_dir)
             )
         return generator
@@ -320,6 +324,4 @@ def create_gradio_interface():
 
 if __name__ == "__main__":
     demo = create_gradio_interface()
-    # Fixed queue configuration for newer Gradio versions
     demo.queue().launch(share=True)
-