anon5 commited on
Commit
aae03c9
·
verified ·
1 Parent(s): 720ff4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ with gr.Blocks() as demo:
8
  def submit(image):
9
  torch.backends.cuda.matmul.allow_tf32 = True
10
  torch.backends.cudnn.allow_tf32 = True
11
- pipe = pipeline("image-classification", model=".\\checkpoint-600")
12
  output = pipe(images=[image])
13
 
14
  result = {}
 
8
  def submit(image):
9
  torch.backends.cuda.matmul.allow_tf32 = True
10
  torch.backends.cudnn.allow_tf32 = True
11
+ pipe = pipeline("image-classification", model="./checkpoint-600")
12
  output = pipe(images=[image])
13
 
14
  result = {}