GIGAParviz commited on
Commit
919e333
·
verified ·
1 Parent(s): 8469f59

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -80,7 +80,7 @@ class ModelWithAttention(nn.Module):
80
 
81
 
82
  device = "cuda"
83
- path = r"Captcha(Best).pt"
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 = r"example"
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)