BMD_graph_obj / app.py
Rathapoom's picture
Update app.py
b92b8d3
raw
history blame contribute delete
696 Bytes
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()