Dane Summers commited on
Commit
4aa1057
·
1 Parent(s): bdfca3e

aider: Modify Gradio interface to accept an image as input, call `predict_bear_type` with the image, and display the prediction.

Browse files

# Aider chat conversation:

USER: Change the gradio interface to accept an image, call `predict_bear_type` and then show the prediction.

ASSISTANT: Sure, I will modify the Gradio interface to accept an image as input, call the `predict_bear_type` function with the image, and then display the prediction.

Here are the changes:

```python
app.py
<<<<<<< HEAD
def greet():
return "Hello, World!"

iface = gr.Interface(fn=greet, inputs=[], outputs="text")
iface.launch()
=======
def greet(img_path):
return predict_bear_type(img_path)

iface = gr.Interface(fn=greet, inputs="image", outputs="text")
iface.launch()
>>>>>>> updated
```

This code changes the `greet` function to accept an image path and return the result of calling `predict_bear_type` with the image path. The Gradio interface is updated to accept an image as input and display text as output. The interface calls the `greet` function when an image is submitted.

Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -11,8 +11,8 @@ def predict_bear_type(img_path):
11
  pred,pred_idx,probs = learner.predict(img)
12
  return pred
13
 
14
- def greet():
15
- return "Hello, World!"
16
 
17
- iface = gr.Interface(fn=greet, inputs=[], outputs="text")
18
  iface.launch()
 
11
  pred,pred_idx,probs = learner.predict(img)
12
  return pred
13
 
14
+ def greet(img_path):
15
+ return predict_bear_type(img_path)
16
 
17
+ iface = gr.Interface(fn=greet, inputs="image", outputs="text")
18
  iface.launch()