Spaces:
Runtime error
Runtime error
| import numpy as np | |
| import gradio as gr | |
| from roboflow import Roboflow | |
| from PIL import Image | |
| def obj(input_img): | |
| #load model | |
| rf = Roboflow(api_key="u5HA4Ou9LROqjjWJYSJV") | |
| project = rf.workspace("modzie-work").project("bmd-result") | |
| model = project.version(1).model | |
| #save to .jpg | |
| img = Image.fromarray(input_img, "RGB") | |
| img.save("image_filename.jpg") | |
| image_save_test = 'image_filename.jpg' | |
| #predict | |
| obj_result = model.predict(image_save_test, confidence=40, overlap=30).json()['predictions'][0]['class'] | |
| if obj_result == "Low Bone": | |
| obj_result = "Osteopenia" | |
| return obj_result | |
| demo = gr.Interface(obj, "image", "text") | |
| demo.launch() |