Spaces:
Sleeping
Sleeping
Create app.py
Browse files
app.py
ADDED
|
@@ -0,0 +1,23 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from ultralytics import YOLO
|
| 2 |
+
from PIL import Image
|
| 3 |
+
onn_xmodel=YOLO("/content/best.onnx")
|
| 4 |
+
import gradio as gr
|
| 5 |
+
def detect_fraction(image):
|
| 6 |
+
results=onn_xmodel.predict([image],stream=False,imgsz=512)
|
| 7 |
+
#results
|
| 8 |
+
res=results[0].plot()
|
| 9 |
+
return Image.fromarray(res)
|
| 10 |
+
|
| 11 |
+
|
| 12 |
+
demo=gr.Interface(fn=detect_fraction,inputs=gr.Image(type='pil',label='Upload your XRAY'),
|
| 13 |
+
outputs=gr.Image(type='pil',label='Results'),
|
| 14 |
+
examples=['image1_0_png.rf.99862308d714bff3f9c410adf5ca93ac.jpg',
|
| 15 |
+
'image1_1035_png.rf.d7493a5653bc3628f7a1b1ec0eb5de85.jpg',
|
| 16 |
+
'image1_1015_png.rf.3b7320c3c40771fa5532bf713a728b83.jpg',
|
| 17 |
+
'image1_1084_png.rf.d9da08b77872f6b4282f2700d216b9b3.jpg',
|
| 18 |
+
'image1_1311_png.rf.27646e931562cc3823c79f79472b8749.jpg',
|
| 19 |
+
'image1_145_png.rf.a69d928d011a93d25a95b7b8380ea25d.jpg'],
|
| 20 |
+
#themes=gr.themes.Soft(primary_hue=gr.themes.colors.amber,secondary_hue=gr.themes.colors.blue),
|
| 21 |
+
title="Detecting bone fractions")
|
| 22 |
+
if __name__=="__main__":
|
| 23 |
+
demo.launch(debug=True)
|