Update app.py

app.py

@@ -1,17 +1,9 @@
-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
@@ -32,9 +24,9 @@ def get_strings(lang):
 
 
 args, gen, enhancer = inferencer()
-strings = get_strings("en")
+strings = get_strings(args.lang)
+
 
-@spaces.GPU(duration=80)
 def infer(
     prompt,
     negative_prompt,
@@ -44,8 +36,7 @@ def infer(
     oriW, oriH,
     sampler,
     size,
-    enhance,
-    progress=gr.Progress(track_tqdm=True)
+    enhance
 ):
     if enhance and enhancer is not None:
         success, enhanced_prompt = enhancer(prompt)
@@ -99,13 +90,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=40, step=1,
+                        label=strings['infer steps'], minimum=1, maximum=200, value=100, 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=False, interactive=True,
+                        label=strings['enhance'], value=enhancer is not None, interactive=True,
                     )
 
                 with gr.Accordion(
@@ -132,12 +123,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,
@@ -176,4 +167,4 @@ def ui():
 
 if __name__ == "__main__":
     interface = ui()
-    interface.launch()
+    interface.launch(server_name="0.0.0.0", server_port=443, share=True)