Dimitre commited on
Commit
4d634a3
·
1 Parent(s): 62c65d0

Adding title and description

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -29,7 +29,11 @@ def predict_fn(image):
29
  scores = postprocess(probs)
30
  return scores
31
 
 
 
32
  iface = gr.Interface(fn=predict_fn,
 
 
33
  inputs=gr.Image(shape=(224, 224)),
34
  outputs=gr.Label(num_top_classes=5),
35
  examples=["apples.jpeg", "banana.jpeg", "car.jpeg"])
 
29
  scores = postprocess(probs)
30
  return scores
31
 
32
+ description = "Using the power of CLIP and a simple small CNN, find images from movies based on what you draw!"
33
+
34
  iface = gr.Interface(fn=predict_fn,
35
+ title="ImageNet classification with mobilenet",
36
+ description="Predict from wich ImageNet class your images belongs",
37
  inputs=gr.Image(shape=(224, 224)),
38
  outputs=gr.Label(num_top_classes=5),
39
  examples=["apples.jpeg", "banana.jpeg", "car.jpeg"])