Frantz103 commited on
Commit
f2e88cc
1 Parent(s): 4db19d8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -14,9 +14,9 @@ def predict(img):
14
 
15
  # Check if the prediction is "Cat" and return the appropriate string
16
  if pred == 'Cat':
17
- return "This is a cat"
18
  else:
19
- return "This is not a cat"
20
 
21
  title = "Is this image of a cat?"
22
  iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512,512)), outputs=gr.outputs.Label(num_top_classes=2), title=title)
 
14
 
15
  # Check if the prediction is "Cat" and return the appropriate string
16
  if pred == 'Cat':
17
+ return "Yes, this is a cat"
18
  else:
19
+ return "No, this is not a cat"
20
 
21
  title = "Is this image of a cat?"
22
  iface = gr.Interface(fn=predict, inputs=gr.inputs.Image(shape=(512,512)), outputs=gr.outputs.Label(num_top_classes=2), title=title)