Spaces:
Sleeping
Sleeping
Commit ·
6a1b0dd
1
Parent(s): 725b653
minor bugs fixed
Browse files- App/app.py +2 -2
- src/training.py +7 -7
App/app.py
CHANGED
|
@@ -17,8 +17,8 @@ from io import BytesIO
|
|
| 17 |
import threading
|
| 18 |
from datetime import timedelta
|
| 19 |
import json
|
| 20 |
-
from dotenv import load_dotenv
|
| 21 |
-
load_dotenv()
|
| 22 |
warnings.filterwarnings("ignore", message=".*QuickGELU mismatch.*")
|
| 23 |
|
| 24 |
|
|
|
|
| 17 |
import threading
|
| 18 |
from datetime import timedelta
|
| 19 |
import json
|
| 20 |
+
# from dotenv import load_dotenv
|
| 21 |
+
# load_dotenv()
|
| 22 |
warnings.filterwarnings("ignore", message=".*QuickGELU mismatch.*")
|
| 23 |
|
| 24 |
|
src/training.py
CHANGED
|
@@ -21,13 +21,13 @@ HF_TOKEN=os.getenv("HF_TOKEN")
|
|
| 21 |
|
| 22 |
MODEL_ID = "PrashantGoyal/findr-clip-ft"
|
| 23 |
|
| 24 |
-
|
| 25 |
-
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
model_path = "model/clip/best.pt"
|
| 31 |
tokenizer=open_clip.get_tokenizer('ViT-B-32')
|
| 32 |
|
| 33 |
def seed_everything(seed=42):
|
|
|
|
| 21 |
|
| 22 |
MODEL_ID = "PrashantGoyal/findr-clip-ft"
|
| 23 |
|
| 24 |
+
model_path = hf_hub_download(
|
| 25 |
+
repo_id=MODEL_ID,
|
| 26 |
+
force_download=True,
|
| 27 |
+
filename="clip/best.pt",
|
| 28 |
+
token=os.getenv("HF_TOKEN")
|
| 29 |
+
)
|
| 30 |
+
# model_path = "model/clip/best.pt"
|
| 31 |
tokenizer=open_clip.get_tokenizer('ViT-B-32')
|
| 32 |
|
| 33 |
def seed_everything(seed=42):
|