Update app.py

app.py

@@ -52,22 +52,27 @@ class EnhancedVideoGenerator:
     def initialize_models(self):
         """Initialize all AI models"""
         try:
-            # Text generation model
-            self.text_generator = pipeline(
-                'text-generation',
-                model='gpt2',
-                device=0 if self.device == "cuda" else -1
-            )
-
-            # Initialize free image generation model
-            self.image_model = AutoModelForCausalLM.from_pretrained(
-                "CompVis/stable-diffusion-v1-4",
-                torch_dtype=torch.float16 if self.device == "cuda" else torch.float32
-            ).to(self.device)
-
+            # Text generation model initialization with error handling
+            try:
+                self.text_generator = pipeline(
+                    'text-generation',
+                    model='gpt2',
+                    device=0 if self.device == "cuda" else -1
+                )
+            except Exception as e:
+                self.logger.warning(f"Text generator initialization failed: {str(e)}")
+                self.text_generator = None
+    
+            # Skip the StableDiffusion model initialization as it requires additional setup
+            self.image_model = None
+            
+            # Initialize stability API attribute
+            self.stability_api = None
+    
         except Exception as e:
             self.logger.error(f"Model initialization failed: {str(e)}")
-            raise
+            # Don't raise exception, allow initialization with degraded functionality
+            pass
 
     def setup_workspace(self):
         """Set up working directory and resources"""