change simple

app.py

@@ -24,7 +24,7 @@ def sanitize_prompt(prompt):
   return sanitized_prompt
 
 #@spaces.GPU
-def process_img2img(image,mask_image,prompt="a person",model_id="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0,num_inference_steps=4):
+def process_img2img(image,mask_image,prompt="a person",strength=0.75,seed=0,num_inference_steps=4):
     print("start process image process_image")
     if image == None:
         print("empty input image returned")
@@ -43,7 +43,7 @@ def process_img2img(image,mask_image,prompt="a person",model_id="black-forest-la
 
 
 @spaces.GPU(duration=180)
-def process_images(image, image2=None,prompt="a girl",inpaint_model="black-forest-labs/FLUX.1-schnell",strength=0.75,seed=0):
+def process_images(image, image2=None,prompt="a girl",strength=0.75,seed=0,progress=gr.Progress(track_tqdm=True)):
     print("start process_images")
     try:
         # I'm not sure when this happen
@@ -65,7 +65,7 @@ def process_images(image, image2=None,prompt="a girl",inpaint_model="black-fores
             mask = image['layers'][0]
 
 
-        output = process_img2img(image["background"],mask,prompt,inpaint_model,strength,seed)
+        output = process_img2img(image["background"],mask,prompt,strength,seed)
     except Exception as e:
         print(f"An error occurred: {e}")
         gr.Error(e)
@@ -117,8 +117,8 @@ with demo_blocks as demo:
                         with gr.Row( equal_height=True):
                             strength = gr.Number(value=0.75, minimum=0, maximum=1.0, step=0.01, label="strength")
                             seed = gr.Number(value=0, minimum=0, step=1, label="seed")
-                        models = ["black-forest-labs/FLUX.1-schnell"]
-                        inpaint_model = gr.Dropdown(label="modes", choices=models, value="black-forest-labs/FLUX.1-schnell") 
+                        #models = ["black-forest-labs/FLUX.1-schnell"]
+                        #inpaint_model = gr.Dropdown(label="modes", choices=models, value="black-forest-labs/FLUX.1-schnell") 
                         id_input=gr.Text(label="Name", visible=False)
                             
                 with gr.Column():
@@ -127,7 +127,7 @@ with demo_blocks as demo:
                     
             
 
-    btn.click(fn=process_images, inputs=[image, image_mask,prompt,inpaint_model,strength,seed], outputs =image_out, api_name='infer')
+    btn.click(fn=process_images, inputs=[image, image_mask,prompt,strength,seed], outputs =image_out, api_name='infer')
     gr.Examples(
                examples=[
                     #["images/00547245_99.jpg", "images/00547245_99_mask.jpg","a beautiful girl,eyes closed",0.8,"images/00547245.jpg"],