Update app.py

app.py

@@ -20,7 +20,11 @@ pipe = DiffusionPipeline.from_pretrained(
     "stabilityai/stable-diffusion-xl-base-1.0",  # Remove torch_dtype and variant for CPU-friendly precision
 )
 
-pipe.to("cpu")  # Use CPU for inference
+# Check if GPU is available and use it for inference
+if torch.cuda.is_available():
+    pipe.to("cuda")  # Use GPU for faster inference
+else:
+    pipe.to("cpu")  # Use CPU if GPU is not available
 
 # Function to generate image from text prompt using Stable Diffusion XL
 def generate_image_from_text(translated_text):
@@ -93,5 +97,5 @@ iface = gr.Interface(
     description="Input Tamil text, and this tool will translate it, generate a short paragraph, and create an image based on the translated text."
 )
 
-# Launch the Gradio app
-iface.launch()
+# Launch the Gradio app with share=True to create a shareable link
+iface.launch(share=True)