Update app.py

app.py

@@ -588,7 +588,8 @@ def sign_message(message, key):
     signed_hash = base64.b64encode(hmac_digest).decode()
     return signed_hash
 
-def run(prompt, radio="model-v2", preset=PRESET_Q, h=1216, w=832, negative_prompt=NEGATIVE_PROMPT, guidance_scale=4.0, randomize_seed=True, seed=42, tpu_inference=False, do_img2img=False, init_image=None, image2image_resize=False, image2image_strength=0, inference_steps=25, progress=gr.Progress(track_tqdm=True)):
+@spaces.GPU
+def run(prompt, radio="model-v2", preset=PRESET_Q, h=1216, w=832, negative_prompt=NEGATIVE_PROMPT, guidance_scale=4.0, randomize_seed=True, seed=42, do_img2img=False, init_image=None, image2image_strength=0, inference_steps=25, progress=gr.Progress(track_tqdm=True)):
     if init_image is None:
         do_img2img = False
         
@@ -597,11 +598,7 @@ def run(prompt, radio="model-v2", preset=PRESET_Q, h=1216, w=832, negative_promp
         init_image = Image.fromarray(init_image)
         init_image = init_image.resize((w, h))
         init_image = np.array(init_image)
-    
-    return zero_inference_api(prompt, radio, preset, h, w, negative_prompt, guidance_scale, randomize_seed, seed, do_img2img, init_image, image2image_strength, inference_steps=inference_steps)
-    
-@spaces.GPU
-def zero_inference_api(prompt, radio="model-v2", preset=PRESET_Q, h=1216, w=832, negative_prompt=NEGATIVE_PROMPT, guidance_scale=4.0, randomize_seed=True, seed=42, do_img2img=False, init_image=None, image2image_strength=0, inference_steps=25, progress=gr.Progress(track_tqdm=True)):
+        
     prompt = prompt.strip() + ", " + preset.strip()
     negative_prompt = negative_prompt.strip() if negative_prompt and negative_prompt.strip() else None
     
@@ -708,7 +705,6 @@ with gr.Blocks(theme=theme) as demo:
             negative_prompt = gr.Textbox(label="Negative prompt", value=NEGATIVE_PROMPT, info="Is only applied for the CFG part, leave blank for unconditional generation")
             seed = gr.Number(label="Seed", value=42, info="Seed for random number generator")
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-            tpu_inference = gr.Checkbox(label="TPU Inference", value=False)
 
             do_img2img = gr.Checkbox(label="Image to Image", value=False)
             init_image = gr.Image(label="Input Image", visible=False)
@@ -731,7 +727,7 @@ with gr.Blocks(theme=theme) as demo:
             prompt.submit
         ],
         fn=run,
-        inputs=[prompt, radio, preset, height, width, negative_prompt, guidance_scale, randomize_seed, seed, tpu_inference, do_img2img, init_image, image2image_resize, image2image_strength, inference_steps],
+        inputs=[prompt, radio, preset, height, width, negative_prompt, guidance_scale, randomize_seed, seed, do_img2img, init_image, image2image_resize, image2image_strength, inference_steps],
         outputs=[output, seed],
         concurrency_limit=1,
     )