Update app.py

app.py

@@ -1,6 +1,6 @@
+# app.py - Final Version with AI-Powered Animations
 # Fix for huggingface_hub compatibility
 import huggingface_hub
-
 if not hasattr(huggingface_hub, 'cached_download'):
     from huggingface_hub import hf_hub_download
     huggingface_hub.cached_download = hf_hub_download
@@ -25,9 +25,6 @@ import torch
 from diffusers import DiffusionPipeline, StableDiffusionPipeline
 import torch
 from transformers import pipeline
-from huggingface_hub import HfFolder, hf_hub_download, model_info
-
-
 
 # Configure Streamlit page
 st.set_page_config(
@@ -232,11 +229,15 @@ def load_models():
     models = {}
     
     try:
-        # Text-to-image model for scene generation
+        # Use CPU-friendly model and float32 precision
         models['text_to_image'] = DiffusionPipeline.from_pretrained(
-            "stabilityai/stable-diffusion-2-1", 
-            torch_dtype=torch.float16
+            "runwayml/stable-diffusion-v1-5", 
+            torch_dtype=torch.float32,  # Use float32 instead of float16
+            safety_checker=None  # Disable safety checker for simplicity
         )
+        # Move to GPU if available, otherwise keep on CPU
+        device = "cuda" if torch.cuda.is_available() else "cpu"
+        models['text_to_image'].to(device)
     except Exception as e:
         st.error(f"Could not load text-to-image model: {str(e)}")
         models['text_to_image'] = None
@@ -290,8 +291,14 @@ def create_story_scene(story, concept, models):
         style = "cartoon style, bright colors, children's book illustration"
         prompt = f"{story}. {CONCEPTS[concept]['name']} concept. {style}"
         
-        # Generate image
-        image = models['text_to_image'](prompt).images[0]
+        # Generate image with fixed settings
+        image = models['text_to_image'](
+            prompt,
+            num_inference_steps=25,  # Fewer steps for faster generation
+            guidance_scale=7.5,
+            height=512,
+            width=512
+        ).images[0]
         
         # Convert to bytes
         buf = io.BytesIO()