Spaces:

bortle
/

astrophotography-object-classifier

Running

ppicazo commited on Nov 24, 2023

Commit

ba318fd

1 Parent(s): 5399b5c

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="bortle/astrophotography-object-classifier-alpha4.2")
 def predict(image):
   predictions = pipeline(image)

 import gradio as gr
 from transformers import pipeline
+pipeline = pipeline(task="image-classification", model="bortle/astrophotography-object-classifier-alpha5")
 def predict(image):
   predictions = pipeline(image)