fix more

app.py

@@ -116,6 +116,7 @@ def infer(prompt: str, negative: str, width: int, height: int, sampler: str, ste
         translate_prompt = prompt
         translate_negative = negative
 
+
     output = replicate.run(
         "stability-ai/stable-diffusion:db21e45d3f7023abc2a46ee38a23973f6dce16bb082a930b0c49861f96d1e5bf",
         input={
@@ -395,7 +396,6 @@ with block:
                 small stable diffusion Demo App. <br />
                 Click <strong>Generate image</strong> Button to generate image. <br />
                 Also Change params to have a try <br />
-                every 512*512 image will cost about 40s. <br />
                 more size may cost more time. <br />
                 It's just a simplified demo, you can use more advanced features optimize image quality <br />
               </p>
@@ -460,9 +460,9 @@ with block:
                             full_width=False,
                         )
 
-                gallery = gr.Image(
+                gallery = gr.Gallery(
                     label="Generated images", show_label=False, elem_id="gallery"
-                ).style(height=512)
+                ).style()
 
         # with gr.Group(elem_id="container-advanced-btns"):
         #     # advanced_button = gr.Button("Advanced options", elem_id="advanced-btn")
@@ -487,6 +487,6 @@ with block:
 block.queue(concurrency_count=80,
             max_size=100).launch(
     max_threads=150,
-    # server_port=6006,
-    # share=True,
+    server_port=6006,
+    share=True,
 )

requirements.txt

@@ -1,4 +1,2 @@
 python-dotenv
-replicate
-ftfy
-py-cpuinfo
+replicate