Update app.py

app.py

@@ -12,15 +12,14 @@ def load_translation_model():
     model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
     return tokenizer, model
 
-# Load Stable Diffusion model
+# ✅ Load Stable Diffusion model for CPU
 @st.cache_resource
 def load_diffusion_model():
     pipe = StableDiffusionPipeline.from_pretrained(
         "CompVis/stable-diffusion-v1-4",
-        torch_dtype=torch.float16 if torch.cuda.is_available() else torch.float32
+        torch_dtype=torch.float32  # ✅ Important for CPU
     )
-    if torch.cuda.is_available():
-        pipe = pipe.to("cuda")
+    pipe = pipe.to("cpu")  # ✅ Force to CPU
     return pipe
 
 # Translate Tamil to English using NLLB