Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -80,7 +80,7 @@ class ModelWithAttention(nn.Module):
|
|
80 |
|
81 |
|
82 |
device = "cuda"
|
83 |
-
path =
|
84 |
|
85 |
|
86 |
from torchvision import transforms
|
@@ -141,7 +141,7 @@ def predict(img= None , link:str = None) -> str:
|
|
141 |
|
142 |
from pathlib import Path
|
143 |
|
144 |
-
path =
|
145 |
|
146 |
list_path = []
|
147 |
list_paths = os.listdir(path)
|
|
|
80 |
|
81 |
|
82 |
device = "cuda"
|
83 |
+
path = "Captcha(Best).pt"
|
84 |
|
85 |
|
86 |
from torchvision import transforms
|
|
|
141 |
|
142 |
from pathlib import Path
|
143 |
|
144 |
+
path = "example"
|
145 |
|
146 |
list_path = []
|
147 |
list_paths = os.listdir(path)
|