anon5 commited on
Commit
bdd35f8
·
verified ·
1 Parent(s): c3cd5d0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -11,7 +11,7 @@ print("Loading model...")
11
  #torch.backends.cuda.matmul.allow_tf32 = True
12
  #torch.backends.cudnn.allow_tf32 = True
13
 
14
- pipe = pipeline("image-classification", model="./checkpoint-600")
15
 
16
  print("Starting webapp...")
17
 
 
11
  #torch.backends.cuda.matmul.allow_tf32 = True
12
  #torch.backends.cudnn.allow_tf32 = True
13
 
14
+ pipe = pipeline("image-classification", model="/checkpoint-600")
15
 
16
  print("Starting webapp...")
17