fix more

app.py

@@ -55,10 +55,10 @@ def infer(prompt: str, negative: str, width: int, height: int, sampler: str, ste
         seed = int(seed)
 
     images = []
-    device = "cpu"
     if torch.cuda.is_available():
-        device = "cuda"
-    generator = torch.Generator(device=device).manual_seed(seed)
+        generator = torch.Generator(device="cuda").manual_seed(seed)
+    else:
+        generator = torch.manual_seed(seed)
     if sampler == "EulerDiscrete":
         pipe.scheduler = EulerDiscreteScheduler.from_config(pipe.scheduler.config)
     elif sampler == "EulerAncestralDiscrete":
@@ -132,6 +132,9 @@ css = """
             margin: auto;
             padding-top: 1.5rem;
         }
+        #prompt-column {
+            min-height: 450px
+        }
         #gallery {
             min-height: 22rem;
             margin-bottom: 15px;
@@ -365,7 +368,7 @@ with block:
     with gr.Group():
         with gr.Box():
             with gr.Row(elem_id="prompt-container").style(mobile_collapse=False, equal_height=True):
-                with gr.Column():
+                with gr.Column(elem_id="prompt-column"):
                     text = gr.Textbox(
                         label="Enter your prompt",
                         show_label=False,