kevin1kevin1k commited on
Commit
64daa59
·
verified ·
1 Parent(s): 0957d9e

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. README.md +4 -2
  2. image_evaluators.py +1 -1
  3. prompt_refiners.py +1 -1
README.md CHANGED
@@ -35,8 +35,10 @@ Given a target image, WeavePrompt automatically generates and refines text promp
35
  uv sync
36
  source .venv/bin/activate
37
  ```
38
- 3. Setup API keys in `.env`
39
- Put `WANDB_API_KEY` and `FAL_KEY`
 
 
40
 
41
  ## Usage
42
 
 
35
  uv sync
36
  source .venv/bin/activate
37
  ```
38
+ 3. Setup `.env`
39
+ Put the following inside `.env`:
40
+ - API keys `WANDB_API_KEY` and `FAL_KEY`
41
+ - Weave project name `WEAVE_PROJECT`
42
 
43
  ## Usage
44
 
image_evaluators.py CHANGED
@@ -14,7 +14,7 @@ from dotenv import load_dotenv
14
  load_dotenv()
15
 
16
  # Weave autopatches OpenAI to log LLM calls to W&B
17
- weave.init("meta-llama")
18
 
19
  class LlamaEvaluator(ImageEvaluator):
20
  """Llama-based image evaluator using W&B Inference."""
 
14
  load_dotenv()
15
 
16
  # Weave autopatches OpenAI to log LLM calls to W&B
17
+ weave.init(os.getenv("WEAVE_PROJECT", "meta-llama"))
18
 
19
  class LlamaEvaluator(ImageEvaluator):
20
  """Llama-based image evaluator using W&B Inference."""
prompt_refiners.py CHANGED
@@ -10,7 +10,7 @@ from weave_prompt import PromptRefiner
10
  load_dotenv()
11
 
12
  # Weave autopatches OpenAI to log LLM calls to W&B
13
- weave.init(project_name="meta-llama")
14
 
15
 
16
  class LlamaPromptRefiner(PromptRefiner):
 
10
  load_dotenv()
11
 
12
  # Weave autopatches OpenAI to log LLM calls to W&B
13
+ weave.init(os.getenv("WEAVE_PROJECT", "meta-llama"))
14
 
15
 
16
  class LlamaPromptRefiner(PromptRefiner):