Spaces:

davanstrien
/

autotrain-blog-example

Runtime error

davanstrien HF Staff commited on Feb 27, 2023

Commit

fdbc942

1 Parent(s): e74bbfb

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -13,7 +13,7 @@ model = AutoModelForImageClassification.from_pretrained(chk_point)
 pipe = pipeline(
     "image-classification",
-    "davanstrien/autotrain-encyclopaedia-illustrations-blog-post-3327992158",
     accelerator="bettertransformer",device=device
 )

 pipe = pipeline(
     "image-classification",
+    chk_point,
     accelerator="bettertransformer",device=device
 )