Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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="
|
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 = {}
|