feat: gradio页面优化：显示中文，直接显示生成图片；去除不必要的依赖

.gitignore

 .idea
 chenyh
+.DS_Store

AuraSR-v2/app.py

-import spaces
 import gradio as gr
-from gradio_imageslider import ImageSlider
 from PIL import Image
-import numpy as np
 from aura_sr import AuraSR
-import torch
 
-# Force CPU usage
-torch.set_default_tensor_type(torch.FloatTensor)
-
-# Override torch.load to always use CPU
-original_load = torch.load
-torch.load = lambda *args, **kwargs: original_load(*args, **kwargs, map_location=torch.device('cpu'))
+# # Force CPU usage
+# torch.set_default_type(torch.FloatTensor)
+# torch.set_default_device('cpu')
+#
+# # Override torch.load to always use CPU
+# original_load = torch.load
+# torch.load = lambda *args, **kwargs: original_load(*args, **kwargs, map_location=torch.device('cpu'))
 
 # Initialize the AuraSR model
-aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2")
+aura_sr = AuraSR.from_pretrained("fal/AuraSR-v2/model.safetensors")
+
 
-# Restore original torch.load
-torch.load = original_load
+# # Restore original torch.load
+# torch.load = original_load
 
 def process_image(input_image):
     if input_image is None:
@@ -29,38 +27,29 @@ def process_image(input_image):
     # Upscale the image using AuraSR
     upscaled_image = process_image_on_gpu(pil_image)
 
-    # Convert result to numpy array if it's not already
-    result_array = np.array(upscaled_image)
+    return upscaled_image
 
-    return [input_image, result_array]
 
-@spaces.GPU
 def process_image_on_gpu(pil_image):
     return aura_sr.upscale_4x(pil_image)
-    
-title = """<h1 align="center">AuraSR-v2 - An open reproduction of the GigaGAN Upscaler from fal.ai</h1>
-<p><center>
-<a href="https://huggingface.co/fal/AuraSR-v2" target="_blank">[AuraSR-v2]</a>
-<a href="https://blog.fal.ai/introducing-aurasr-an-open-reproduction-of-the-gigagan-upscaler-2/" target="_blank">[Blog Post]</a>
-<a href="https://huggingface.co/fal-ai/AuraSR" target="_blank">[v1 Model Page]</a>
-</center></p>
-"""
+
+
+title = """<h1 align="center">AuraSR-v2：一款基于GAN图像修复工具，可从低分辨率图片生成高分辨率图片</h1>"""
 
 with gr.Blocks() as demo:
-    
     gr.HTML(title)
-    
+
     with gr.Row():
         with gr.Column(scale=1):
-            input_image = gr.Image(label="Input Image", type="numpy")
-            process_btn = gr.Button("Upscale Image")
+            input_image = gr.Image(label="输入图片", type="numpy")
+            process_btn = gr.Button("生成")
         with gr.Column(scale=1):
-            output_slider = ImageSlider(label="Before / After", type="numpy")
+            gallery = gr.Image(label="生成图片")
 
     process_btn.click(
         fn=process_image,
         inputs=[input_image],
-        outputs=output_slider
+        outputs=gallery
     )
 
     # Add examples
@@ -70,9 +59,9 @@ with gr.Blocks() as demo:
             "image3.png"
         ],
         inputs=input_image,
-        outputs=output_slider,
+        outputs=gallery,
         fn=process_image,
         cache_examples=True
     )
 
-demo.launch(debug=True)
\ No newline at end of file
+demo.launch(server_name='0.0.0.0', share=True)

AuraSR-v2/requirements.txt

 spaces
-aura-sr
-gradio-imageslider
\ No newline at end of file
+aura-sr
\ No newline at end of file