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)
|