suvadityamuk commited on
Commit
b6b4764
·
1 Parent(s): ad11fe1

chore: made optims

Browse files

Signed-off-by: Suvaditya Mukherjee <[email protected]>

Files changed (1) hide show
  1. app.py +13 -1
app.py CHANGED
@@ -125,10 +125,22 @@ if __name__ == "__main__":
125
  parallel=0,
126
  )
127
 
128
- wandb.init(project="resume-rag", name="zerogpu-run")
 
 
129
 
130
  model_name = "Qwen/Qwen2.5-3B-Instruct"
131
 
 
 
 
 
 
 
 
 
 
 
132
  @spaces.GPU
133
  def rag_process(message, chat_history):
134
  # Append current user message to chat history
 
125
  parallel=0,
126
  )
127
 
128
+ wandb.login(
129
+ api_key=os.getenv("WANDB_API_KEY")
130
+ )
131
 
132
  model_name = "Qwen/Qwen2.5-3B-Instruct"
133
 
134
+ wandb.init(
135
+ project="resume-rag",
136
+ name="zerogpu-run",
137
+ save_code=True,
138
+ config={
139
+ "model_name": model_name,
140
+ "resume_url": RESUME_URL
141
+ }
142
+ )
143
+
144
  @spaces.GPU
145
  def rag_process(message, chat_history):
146
  # Append current user message to chat history