Spaces:

mvaloatto
/

bald-or-not

Runtime error

mvaloatto commited on Jan 28, 2023

Commit

a39bdf9

1 Parent(s): e1baead

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,7 +1,7 @@
 import gradio as gr
 from transformers import pipeline
-pipeline = pipeline(task="image-classification", model="mvaloatto/baldhead-spotter")
 def predict(image):
   predictions = pipeline(image)

 import gradio as gr
 from transformers import pipeline
+pipeline = pipeline(task="image-classification", model="mvaloatto/bald-or-not")
 def predict(image):
   predictions = pipeline(image)