Update app.py

app.py

@@ -1,9 +1,17 @@
+import os
+from huggingface_hub import snapshot_download
+import spaces
+import torch
+torch.jit.script = lambda f: f
+os.makedirs("/home/user/app/ckpts", exist_ok=True)
+snapshot_download(repo_id="Tencent-Hunyuan/HunyuanDiT-v1.1", local_dir="/home/user/app/ckpts")
+
 import gradio as gr
 import pandas as pd
 from pathlib import Path
 from PIL import Image
 import sys
-sys.path.insert(0, str(Path(__file__).parent.parent))
+#sys.path.insert(0, str(Path(__file__).parent.parent))
 
 from hydit.constants import SAMPLER_FACTORY
 from sample_t2i import inferencer
@@ -24,9 +32,9 @@ def get_strings(lang):
 
 
 args, gen, enhancer = inferencer()
-strings = get_strings(args.lang)
-
+strings = get_strings("en")
 
+@spaces.GPU(duration=80)
 def infer(
     prompt,
     negative_prompt,
@@ -36,7 +44,8 @@ def infer(
     oriW, oriH,
     sampler,
     size,
-    enhance
+    enhance,
+    progress=gr.Progress(track_tqdm=True)
 ):
     if enhance and enhancer is not None:
         success, enhanced_prompt = enhancer(prompt)
@@ -90,13 +99,13 @@ def ui():
                 prompt = gr.Textbox(label=strings['prompt'], value=strings['default prompt'], lines=3)
                 with gr.Row():
                     infer_steps = gr.Slider(
-                        label=strings['infer steps'], minimum=1, maximum=200, value=100, step=1,
+                        label=strings['infer steps'], minimum=1, maximum=200, value=40, step=1,
                     )
                     seed = gr.Number(
                         label=strings['seed'], minimum=-1, maximum=1_000_000_000, value=1, step=1, precision=0,
                     )
                     enhance = gr.Checkbox(
-                        label=strings['enhance'], value=enhancer is not None, interactive=True,
+                        label=strings['enhance'], value=False, interactive=True,
                     )
 
                 with gr.Accordion(
@@ -123,12 +132,12 @@ def ui():
                 with gr.Row():
                     advanced_button = gr.Button(strings['run'])
             with gr.Column():
-                default_img = Image.open(ROOT / 'app/default.png')
+                #default_img = Image.open(ROOT / 'app/default.png')
                 output_img = gr.Image(
                     label=strings['generated image'],
                     interactive=False,
                     format='png',
-                    value=default_img,
+                    #value=default_img,
                 )
             advanced_button.click(
                 fn=infer,
@@ -167,4 +176,4 @@ def ui():
 
 if __name__ == "__main__":
     interface = ui()
-    interface.launch(server_name="0.0.0.0", server_port=443, share=True)
+    interface.launch()