Thiago Hersan commited on
Commit
71cdb5c
·
1 Parent(s): 8358707

add model descriptions

Browse files
Files changed (2) hide show
  1. README.md +1 -1
  2. app.py +7 -4
README.md CHANGED
@@ -1,5 +1,5 @@
1
  ---
2
- title: PUIL ML Model Experiments
3
  emoji: 🌴🦤
4
  colorFrom: blue
5
  colorTo: red
 
1
  ---
2
+ title: PUIL3300 Model "Forensics" Experiments
3
  emoji: 🌴🦤
4
  colorFrom: blue
5
  colorTo: red
app.py CHANGED
@@ -58,13 +58,14 @@ def run_clip(files, word0, word1=""):
58
 
59
 
60
  examples = [
61
- ("colorful painted portrait of person", "gray painted portrait of person"),
62
- ("painted portrait of young person", "painted portrait of old person"),
63
  ]
64
 
65
  with gr.Blocks() as demo:
66
  gr.Interface(
67
  title="Object Detection",
 
68
  api_name="object",
69
  fn=run_detr,
70
  inputs=gr.Image(type="pil"),
@@ -74,6 +75,7 @@ with gr.Blocks() as demo:
74
 
75
  gr.Interface(
76
  title="Zero-Shot Object Detection",
 
77
  api_name="zero",
78
  fn=run_owl,
79
  inputs=[gr.Image(type="pil"), gr.Textbox(label="Object", show_label=True)],
@@ -83,11 +85,12 @@ with gr.Blocks() as demo:
83
 
84
  gr.Interface(
85
  title="Contrastive Embedding",
 
86
  api_name="clip",
87
  fn=run_clip,
88
  inputs=[gr.File(file_count="multiple"),
89
- gr.Textbox(label="1st Description", show_label=True),
90
- gr.Textbox(label="2nd Description", show_label=True)],
91
  outputs=gr.Image(format="jpeg"),
92
  flagging_mode="never",
93
  )
 
58
 
59
 
60
  examples = [
61
+ ("painted portrait young person", "painted portrait old person"),
62
+ ("painted portrait happy person", "painted portrait worried person"),
63
  ]
64
 
65
  with gr.Blocks() as demo:
66
  gr.Interface(
67
  title="Object Detection",
68
+ description="[DETR](https://huggingface.co/facebook/detr-resnet-50) model from facebook (2020), trained on [COCO 2017](https://github.com/amikelive/coco-labels/blob/master/coco-labels-2014_2017.txt) dataset and labels.",
69
  api_name="object",
70
  fn=run_detr,
71
  inputs=gr.Image(type="pil"),
 
75
 
76
  gr.Interface(
77
  title="Zero-Shot Object Detection",
78
+ description="[OWLv2](https://huggingface.co/google/owlv2-large-patch14-ensemble) model from google (2023).",
79
  api_name="zero",
80
  fn=run_owl,
81
  inputs=[gr.Image(type="pil"), gr.Textbox(label="Object", show_label=True)],
 
85
 
86
  gr.Interface(
87
  title="Contrastive Embedding",
88
+ description="[CLIP](https://huggingface.co/openai/clip-vit-large-patch14) model from openai (2021).",
89
  api_name="clip",
90
  fn=run_clip,
91
  inputs=[gr.File(file_count="multiple"),
92
+ gr.Textbox(label="1st Descriptor", show_label=True),
93
+ gr.Textbox(label="2nd Descriptor", show_label=True)],
94
  outputs=gr.Image(format="jpeg"),
95
  flagging_mode="never",
96
  )