fix some bugs

app.py

@@ -134,7 +134,7 @@ def build_demo():
 
     with gr.Blocks(css=css) as demo:
         with gr.Column(elem_id="col-container"):
-            gr.Markdown("""<h1 style="text-align: center; font-size: 32px;"><b>UltraFusion for HDR 📸✨</b></h1>""")
+            gr.Markdown("""<h1 style="text-align: center; font-size: 32px;"><b>UltraFusion HDR 📸✨</b></h1>""")
             # gr.Markdown("""<h1 style="text-align: center; font-size: 32px;"><b>OpenImagingLab</b></h1>""")
             gr.Markdown("""<h1 style="text-align: center; font-size: 24px;"><b>How do I use it?</b></h1>""")
             with gr.Row():

requirements.txt

@@ -8,4 +8,5 @@ pillow
 einops
 scipy
 ftfy
+torchvision==0.19.1
 pytorch_lightning==2.4