lisonallen commited on
Commit
10428c0
·
1 Parent(s): 7d0da85

Fix dependency conflicts and update UI for compatibility with Gradio 3.19.1

Browse files
Files changed (2) hide show
  1. app.py +21 -45
  2. requirements.txt +2 -2
app.py CHANGED
@@ -111,7 +111,6 @@ def generate_ai_image(prompt, seed=None):
111
  model_id,
112
  torch_dtype=torch_dtype,
113
  use_auth_token=False, # 明确不使用认证
114
- revision="main", # 使用主分支
115
  safety_checker=None, # 禁用安全检查器
116
  )
117
  pipe = pipe.to(device)
@@ -249,50 +248,28 @@ def generate_image(prompt):
249
  logger.warning("Using rule-based image generation")
250
  return generate_rule_based_image(prompt)
251
 
252
- # 创建Gradio界面
253
  def create_demo():
254
- with gr.Blocks(title="Text to Image Generator") as demo:
255
- gr.Markdown("# Text to Image Generator")
256
- gr.Markdown("Enter a text description to generate an image.")
257
-
258
- with gr.Row():
259
- with gr.Column(scale=3):
260
- # 输入区域
261
- prompt_input = gr.Textbox(
262
- label="Prompt",
263
- placeholder="Describe the image you want, e.g.: a cute cat, sunset over mountains...",
264
- lines=2
265
- )
266
- generate_button = gr.Button("Generate Image", variant="primary")
267
-
268
- # 示例
269
- gr.Examples(
270
- examples=[
271
- "a cute cat sitting on a windowsill",
272
- "beautiful sunset over mountains",
273
- "an astronaut riding a horse in space",
274
- "a fantasy castle on a floating island"
275
- ],
276
- inputs=prompt_input
277
- )
278
-
279
- # 输出区域
280
- with gr.Column(scale=5):
281
- output_image = gr.Image(label="Generated Image", type="pil")
282
-
283
- # 绑定按钮事件
284
- generate_button.click(
285
- fn=generate_image,
286
- inputs=prompt_input,
287
- outputs=output_image
288
- )
289
-
290
- # 也绑定Enter键提交
291
- prompt_input.submit(
292
- fn=generate_image,
293
- inputs=prompt_input,
294
- outputs=output_image
295
- )
296
 
297
  return demo
298
 
@@ -305,7 +282,6 @@ if __name__ == "__main__":
305
  logger.info("Starting Gradio interface...")
306
  demo.launch(
307
  server_name="0.0.0.0",
308
- show_api=False,
309
  share=False
310
  )
311
  except Exception as e:
 
111
  model_id,
112
  torch_dtype=torch_dtype,
113
  use_auth_token=False, # 明确不使用认证
 
114
  safety_checker=None, # 禁用安全检查器
115
  )
116
  pipe = pipe.to(device)
 
248
  logger.warning("Using rule-based image generation")
249
  return generate_rule_based_image(prompt)
250
 
251
+ # 为旧版 gradio 创建界面
252
  def create_demo():
253
+ # 使用 Interface 替代 Blocks (兼容3.19.1)
254
+ demo = gr.Interface(
255
+ fn=generate_image,
256
+ inputs=gr.Textbox(
257
+ label="Prompt",
258
+ placeholder="Describe the image you want, e.g.: a cute cat, sunset over mountains...",
259
+ lines=2
260
+ ),
261
+ outputs=gr.Image(label="Generated Image", type="pil"),
262
+ title="Text to Image Generator",
263
+ description="Enter a text description to generate an image.",
264
+ examples=[
265
+ "a cute cat sitting on a windowsill",
266
+ "beautiful sunset over mountains",
267
+ "an astronaut riding a horse in space",
268
+ "a fantasy castle on a floating island"
269
+ ],
270
+ allow_flagging=False,
271
+ cache_examples=False
272
+ )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
273
 
274
  return demo
275
 
 
282
  logger.info("Starting Gradio interface...")
283
  demo.launch(
284
  server_name="0.0.0.0",
 
285
  share=False
286
  )
287
  except Exception as e:
requirements.txt CHANGED
@@ -1,8 +1,8 @@
1
  accelerate==0.15.0
2
  diffusers==0.10.2
3
- huggingface-hub==0.11.1
4
  torch==1.13.1
5
  transformers==4.25.1
6
  safetensors==0.3.1
7
- gradio==3.24.1
8
  Pillow==9.5.0
 
1
  accelerate==0.15.0
2
  diffusers==0.10.2
3
+ huggingface-hub==0.13.0
4
  torch==1.13.1
5
  transformers==4.25.1
6
  safetensors==0.3.1
7
+ gradio==3.19.1
8
  Pillow==9.5.0