new commit

README.md

@@ -4,7 +4,8 @@ emoji: 🚀
 colorFrom: yellow
 colorTo: pink
 sdk: gradio
-sdk_version: 5.6.0
+python_version: 3.9
+sdk_version: 4.44.0
 app_file: app.py
 pinned: false
 ---

app.py

@@ -69,7 +69,7 @@ def image_preprocess(image: Image, mode="RGB", return_orig=False):
         return out_img
 
 
-def infer(prompt, image):
+def infer(image):
     source = image["background"].convert("RGB").resize(size)
 
     mask = image["layers"][0]
@@ -161,13 +161,9 @@ prefix = ""
 title = f"""
 			<div class="main-div">
 			  <div>
-				<h1>Small Stable Diffusion V0</h1>
+				<h1>Lama model</h1>
 			  </div>
-			  <p>
-			   Demo for <a href="https://huggingface.co/OFA-Sys/small-stable-diffusion-v0">Small Stable Diffusion V0</a> Stable Diffusion model.<br>
-			   {"Add the following tokens to your prompts for the model to work properly: <b>prefix</b>" if prefix else ""}
-			  </p>
-			  Running on {"<b>GPU 🔥</b>" if torch.cuda.is_available() else f"<b>CPU 🥶</b>. For faster inference it is recommended to <b>upgrade to GPU in <a href='https://huggingface.co/spaces/akhaliq/small-stable-diffusion-v0/settings'>Settings</a></b>"} after duplicating the space<br><br>
+			  Running on {"<b>GPU 🔥</b>" if torch.cuda.is_available() else "<b>CPU 🥶</b>"} <br><br>
 			  <a style="display:inline-block" href="https://huggingface.co/spaces/akhaliq/small-stable-diffusion-v0?duplicate=true"><img src="https://bit.ly/3gLdBN6" alt="Duplicate Space"></a>
 			</div>
 		"""
@@ -176,12 +172,6 @@ with gr.Blocks(css=css) as demo:
     gr.HTML(title)
     with gr.Row():
         with gr.Row():
-            with gr.Column():
-                prompt = gr.Textbox(
-                    label="Prompt",
-                    info="Describe what to inpaint the mask with",
-                    lines=3,
-                )
             with gr.Column():
                 with gr.Row():
                     with gr.Column():
@@ -219,7 +209,7 @@ with gr.Blocks(css=css) as demo:
         outputs=use_as_input_button,
     ).then(
         fn=infer,
-        inputs=[prompt, input_image],
+        inputs=[input_image],
         outputs=result,
     ).then(
         fn=lambda: gr.update(visible=True),
@@ -227,22 +217,22 @@ with gr.Blocks(css=css) as demo:
         outputs=use_as_input_button,
     )
 
-    prompt.submit(
-        fn=clear_result,
-        inputs=None,
-        outputs=result,
-    ).then(
-        fn=lambda: gr.update(visible=False),
-        inputs=None,
-        outputs=use_as_input_button,
-    ).then(
-        fn=infer,
-        inputs=[prompt, input_image],
-        outputs=result,
-    ).then(
-        fn=lambda: gr.update(visible=True),
-        inputs=None,
-        outputs=use_as_input_button,
-    )
+    # prompt.submit(
+    #     fn=clear_result,
+    #     inputs=None,
+    #     outputs=result,
+    # ).then(
+    #     fn=lambda: gr.update(visible=False),
+    #     inputs=None,
+    #     outputs=use_as_input_button,
+    # ).then(
+    #     fn=infer,
+    #     inputs=[prompt, input_image],
+    #     outputs=result,
+    # ).then(
+    #     fn=lambda: gr.update(visible=True),
+    #     inputs=None,
+    #     outputs=use_as_input_button,
+    # )
 
 demo.launch()

requirements.txt

@@ -1,6 +1,7 @@
 pyyaml
 tqdm
-numpy
+mxnet-mkl==1.6.0
+numpy==1.23.1
 easydict==1.9.0
 scikit-image==0.17.2
 scikit-learn==0.24.2