davanstrien HF Staff commited on
Commit
e74bbfb
·
1 Parent(s): 8eeaa6b

update checkpoint

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -1,14 +1,13 @@
1
  from pathlib import Path
2
  import gradio as gr
3
  import torch
4
- # from transformers import pipeline
5
  from transformers import AutoModelForImageClassification
6
  import shutil
7
  from optimum.pipelines import pipeline
8
 
9
  device = 1 if torch.cuda.is_available() else "cpu"
10
 
11
- chk_point = "davanstrien/autotrain-encyclopaedia-illustrations-blog-post-3327992158"
12
 
13
  model = AutoModelForImageClassification.from_pretrained(chk_point)
14
 
 
1
  from pathlib import Path
2
  import gradio as gr
3
  import torch
 
4
  from transformers import AutoModelForImageClassification
5
  import shutil
6
  from optimum.pipelines import pipeline
7
 
8
  device = 1 if torch.cuda.is_available() else "cpu"
9
 
10
+ chk_point = "davanstrien/autotrain-ia-useful-covers-3665397856"
11
 
12
  model = AutoModelForImageClassification.from_pretrained(chk_point)
13