Spaces:

dqnguyen
/

Diabetic-Foot-Ulcer-Image-Classification

Sleeping

dqnguyen commited on Aug 25, 2023

Commit

0114414

•

1 Parent(s): 31cba7e

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@
 import gradio as gr
 from transformers import pipeline
-pipeline = pipeline(task="image-classification", model="dqnguyen/vit-base_diabetic_ulcer_image_classification")
 def predict(image):
     predictions = pipeline(image)

 import gradio as gr
 from transformers import pipeline
+pipeline = pipeline(task="image-classification", model="dqnguyen/Diabetic_Foot_Ulcer_Image_Classification")
 def predict(image):
     predictions = pipeline(image)