Update app.py

app.py

@@ -3,8 +3,9 @@ import numpy as np
 import random
 from diffusers import DiffusionPipeline
 import torch
+import spaces
 
-device = "cuda" if torch.cuda.is_available() else "cpu"
+device = "cuda" 
 
 if torch.cuda.is_available():
     torch.cuda.max_memory_allocated(device=device)
@@ -50,10 +51,7 @@ css="""
 }
 """
 
-if torch.cuda.is_available():
-    power_device = "GPU"
-else:
-    power_device = "CPU"
+power_device = "GPU"
 
 with gr.Blocks(css=css) as demo:
     
@@ -65,27 +63,12 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             
-            prompt = gr.Text(
-                label="Prompt",
-                show_label=False,
-                max_lines=1,
-                placeholder="Enter your prompt",
-                container=False,
-            )
-            
             run_button = gr.Button("Run", scale=0)
         
         result = gr.Image(label="Result", show_label=False)
 
         with gr.Accordion("Advanced Settings", open=False):
             
-            negative_prompt = gr.Text(
-                label="Negative prompt",
-                max_lines=1,
-                placeholder="Enter a negative prompt",
-                visible=False,
-            )
-            
             seed = gr.Slider(
                 label="Seed",
                 minimum=0,
@@ -139,7 +122,7 @@ with gr.Blocks(css=css) as demo:
 
     run_button.click(
         fn = infer,
-        inputs = [prompt, negative_prompt, seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
+        inputs = [seed, randomize_seed, width, height, guidance_scale, num_inference_steps],
         outputs = [result]
     )