guychuk commited on
Commit
c6bd1a4
Β·
verified Β·
1 Parent(s): 450f958

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from optimum.pipelines import pipeline as onnx_pipeline
4
 
5
  # Load Models
6
  clean_pipe = pipeline("image-classification", model="WinKawaks/vit-small-patch16-224")
7
- mal_pipe = onnx_pipeline("image-classification", model="WinKawaks/vit-small-patch16-224", accelerator="ort")
8
 
9
  # Interface Functions
10
 
 
4
 
5
  # Load Models
6
  clean_pipe = pipeline("image-classification", model="WinKawaks/vit-small-patch16-224")
7
+ mal_pipe = onnx_pipeline("image-classification", model="onnx/model.onnx", accelerator="ort")
8
 
9
  # Interface Functions
10