| import io |
| import os |
| import requests |
| from PIL import Image |
| from model import get_caption_model, generate_caption |
| import gradio as gr |
|
|
| def get_model(): |
| return get_caption_model() |
| caption_model = get_model() |
| def predict(): |
| captions = [] |
| pred_caption = generate_caption('tmp.jpg', caption_model) |
| captions.append(pred_caption) |
| return captions; |
| def launch(inputs): |
| img = Image.open(requests.get(inputs, stream=True).raw) |
| img = img.convert('RGB') |
| img.save('tmp.jpg') |
| predictedCaption=predict() |
| os.remove('tmp.jpg') |
| return predictedCaption[0] |
| iface = gr.Interface(launch, inputs="text", outputs="text",) |
| iface.launch(debug=True,) |
|
|
|
|
|
|
|
|
|
|
