Update app.py

app.py

@@ -5,8 +5,14 @@ from PIL import Image
 
 
 model_id = "CompVis/stable-diffusion-v1-4"
-pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
-pipe = pipe.to("cuda" if torch.cuda.is_available() else "cpu")
+
+
+if torch.cuda.is_available():
+    pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id, torch_dtype=torch.float16)
+    pipe = pipe.to("cuda")
+else:
+    pipe = StableDiffusionImg2ImgPipeline.from_pretrained(model_id)
+    pipe = pipe.to("cpu")
 
 def stylize_image(input_image, prompt):
     input_image = input_image.convert("RGB")
@@ -18,14 +24,14 @@ def stylize_image(input_image, prompt):
 iface = gr.Interface(
     fn=stylize_image, 
     inputs=[
-        gr.Image(type="pil", label="Upload your image"), 
-        gr.Textbox(placeholder="Enter the art style... (e.g., Van Gogh style)", label="Art Style", lines=1) 
+        gr.Image(type="pil", label="Upload your image"),
+        gr.Textbox(placeholder="Enter the art style... (e.g., Van Gogh style)", label="Art Style", lines=1)
     ],
-    outputs=gr.Image(label="Stylized Image"),  # English label for output
-    title="Art and Style Transfer Demo",  # English title
-    description="This demo uses the Stable Diffusion model to transform an image into a specified art style. Upload an image and enter a style prompt to get started.",  
+    outputs=gr.Image(label="Stylized Image"),
+    title="Art and Style Transfer Demo",
+    description="This demo uses the Stable Diffusion model to transform an image into a specified art style. Upload an image and enter a style prompt to get started.",
     examples=[
-        ["ben-grayland-gD-TjgDW0so-unsplash.jpg", "Van Gogh style"],  
+        ["ben-grayland-gD-TjgDW0so-unsplash.jpg", "Van Gogh style"],
     ]
 )