Update app.py

app.py

@@ -15,13 +15,12 @@ text_tokenizer = AutoTokenizer.from_pretrained(text_generation_model_name)
 text_model = AutoModelForCausalLM.from_pretrained(text_generation_model_name)
 text_generator = pipeline("text-generation", model=text_model, tokenizer=text_tokenizer)
 
-# Load the Stable Diffusion XL Model for Image Generation
+# Load the Stable Diffusion XL Model for Image Generation in full precision (fp32)
 pipe = DiffusionPipeline.from_pretrained(
-    "stabilityai/stable-diffusion-xl-base-1.0", 
-    torch_dtype=torch.float16,
-    variant="fp16",
+    "stabilityai/stable-diffusion-xl-base-1.0",  # Remove torch_dtype and variant for CPU-friendly precision
 )
-pipe.to("cuda")  # Send the model to GPU for faster image generation
+
+pipe.to("cpu")  # Use CPU for inference
 
 # Function to generate image from text prompt using Stable Diffusion XL
 def generate_image_from_text(translated_text):