PrashantGoyal commited on
Commit
6a1b0dd
·
1 Parent(s): 725b653

minor bugs fixed

Browse files
Files changed (2) hide show
  1. App/app.py +2 -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
- # 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):
 
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):