Spaces:
Runtime error
Runtime error
Upload folder using huggingface_hub
Browse files- README.md +4 -2
- image_evaluators.py +1 -1
- 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
|
| 39 |
-
Put
|
|
|
|
|
|
|
| 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(
|
| 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):
|