Update app.py

app.py

@@ -14,16 +14,19 @@ def translate_to_english(text):
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
 
-# Load Stable Diffusion
-pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", torch_dtype=torch.float16)
-pipe = pipe.to("cuda")
+# Load Stable Diffusion on CPU
+pipe = StableDiffusionPipeline.from_pretrained(
+    "runwayml/stable-diffusion-v1-5",
+    torch_dtype=torch.float32  # Use float32 for CPU
+)
+pipe = pipe.to("cpu")  # Force CPU mode
 
 def generate_image(prompt):
     if not prompt.isascii():  # If non-English
         prompt = translate_to_english(prompt)
     
-    with torch.autocast("cuda"):
-        image = pipe(prompt).images[0]
+    # Generate image (no autocast on CPU)
+    image = pipe(prompt).images[0]
     return image
 
 # Gradio Interface
@@ -31,8 +34,8 @@ app = gr.Interface(
     fn=generate_image,
     inputs=gr.Textbox(label="Enter prompt (any language)"),
     outputs=gr.Image(label="Generated Image"),
-    title="🌍 Multilingual Text-to-Image Generator",
-    description="Type in **English, हिंदी, मराठी, Deutsch, etc.** and get an image!"
+    title="🌍 Multilingual Text-to-Image Generator (CPU Mode)",
+    description="Type in **English, हिंदी, मराठी, Deutsch, etc.** and get an image! (Slower on CPU)"
 )
 
 app.launch()