Update app.py

app.py

@@ -1,34 +1,34 @@
-import gradio as gr
-from diffusers import StableDiffusion3Pipeline
-import torch
-import os
-from huggingface_hub import login
+# import gradio as gr
+# from diffusers import StableDiffusion3Pipeline
+# import torch
+# import os
+# from huggingface_hub import login
 
-# 通过环境变量获取 Token
-hf_token = os.getenv("HF_TOKEN")
+# # 通过环境变量获取 Token
+# hf_token = os.getenv("HF_TOKEN")
 
-# 使用 Hugging Face Token 登录
-login(token=hf_token)
+# # 使用 Hugging Face Token 登录
+# login(token=hf_token)
 
-pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16)
-pipe.load_lora_weights("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", weight_name="Photorealistic-SD3.5-Large-LoRA.safetensors")
-pipe.fuse_lora(lora_scale=1.0)
+# pipe = StableDiffusion3Pipeline.from_pretrained("stabilityai/stable-diffusion-3.5-large", torch_dtype=torch.bfloat16)
+# pipe.load_lora_weights("prithivMLmods/SD3.5-Large-Photorealistic-LoRA", weight_name="Photorealistic-SD3.5-Large-LoRA.safetensors")
+# pipe.fuse_lora(lora_scale=1.0)
 
-# 定义图像生成函数
-def generate_image(prompt):
-    print(prompt)
-    return pipe(prompt).images[0]
+# # 定义图像生成函数
+# def generate_image(prompt):
+#     print(prompt)
+#     return pipe(prompt).images[0]
 
-# 创建 Gradio 界面
-iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
+# # 创建 Gradio 界面
+# iface = gr.Interface(fn=generate_image, inputs="text", outputs="image")
 
-# 启动界面
-iface.launch()
+# # 启动界面
+# iface.launch()
 
 
-# import gradio as gr
+import gradio as gr
 
-# gr.load("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA").launch()
+gr.load("models/prithivMLmods/SD3.5-Large-Photorealistic-LoRA").launch()