TieuDaoChanNhan commited on
Commit
1b291e6
·
1 Parent(s): 15223ef
Files changed (2) hide show
  1. app.py +24 -12
  2. requirements.txt +6 -3
app.py CHANGED
@@ -290,19 +290,31 @@ if __name__ == "__main__":
290
  # except Exception:
291
  # print("gradio:", gr.__version__)
292
 
293
- demo = gr.Interface(
294
- fn=safe_gradio_wrapper,
295
- inputs=gr.Image(type="pil", label="Upload Image"),
296
- outputs=gr.Image(type="pil", label="Generated Emoji"),
297
- title="DTN Face-to-Emoji Generator"
298
- )
299
 
300
- # demo = gr.Interface(
301
- # fn=generate_emoji_fixed,
302
- # inputs=gr.Image(type="pil", label="Upload Image"),
303
- # outputs=gr.Image(type="pil", label="Generated Emoji"),
304
- # title="DTN Face-to-Emoji Generator"
305
- # )
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
306
 
307
 
308
  print("✅ Gradio interface created!")
 
290
  # except Exception:
291
  # print("gradio:", gr.__version__)
292
 
293
+ with gr.Blocks(title="DTN Face-to-Emoji Generator") as demo:
294
+ gr.Markdown("# DTN Face-to-Emoji Generator")
 
 
 
 
295
 
296
+ with gr.Row():
297
+ input_image = gr.Image(label="Upload Image", type="pil")
298
+ output_image = gr.Image(label="Generated Emoji", type="pil")
299
+
300
+ submit_btn = gr.Button("Generate Emoji", variant="primary")
301
+ submit_btn.click(
302
+ fn=safe_gradio_wrapper,
303
+ inputs=[input_image],
304
+ outputs=[output_image]
305
+ )
306
+
307
+ # with gr.Blocks() as demo:
308
+ # with gr.Row():
309
+ # input_image = gr.Image(label="Upload Image")
310
+ # output_image = gr.Image(label="Generated Emoji")
311
+
312
+ # submit_btn = gr.Button("Generate Emoji")
313
+ # submit_btn.click(
314
+ # fn=generate_emoji_fixed,
315
+ # inputs=[input_image],
316
+ # outputs=[output_image]
317
+ # )
318
 
319
 
320
  print("✅ Gradio interface created!")
requirements.txt CHANGED
@@ -1,11 +1,14 @@
1
  torch
2
- torchvision
3
  pillow
4
  huggingface_hub
5
  matplotlib
6
  opencv-python
7
  facenet-pytorch
8
  requests
9
- fastapi
 
 
 
 
10
  uvicorn
11
- numpy
 
1
  torch
2
+ torchvision
3
  pillow
4
  huggingface_hub
5
  matplotlib
6
  opencv-python
7
  facenet-pytorch
8
  requests
9
+ numpy
10
+ pydantic==2.10.6
11
+ gradio==4.44.0
12
+ gradio-client==1.3.0
13
+ fastapi==0.112.4
14
  uvicorn