| import gradio as gr | |
| import spaces | |
| from core import IQA | |
| metric = IQA("nima-vgg16-ava") | |
| def greet(image): | |
| return metric(image).item() | |
| demo = gr.Interface(fn=greet, inputs=gr.Image(format="png", image_mode="RGB", type="pil"), outputs=gr.Textbox()) | |
| demo.launch() | |