Update main.py
Browse files
main.py
CHANGED
|
@@ -6,7 +6,7 @@ import prompt_style
|
|
| 6 |
|
| 7 |
|
| 8 |
model_id = "failspy/Meta-Llama-3-8B-Instruct-abliterated-v3"
|
| 9 |
-
client = InferenceClient(token=HF_TOKEN, model=model_id)
|
| 10 |
|
| 11 |
class Item(BaseModel):
|
| 12 |
prompt: str
|
|
|
|
| 6 |
|
| 7 |
|
| 8 |
model_id = "failspy/Meta-Llama-3-8B-Instruct-abliterated-v3"
|
| 9 |
+
client = InferenceClient(token=os.getenv('HF_TOKEN'), model=model_id)
|
| 10 |
|
| 11 |
class Item(BaseModel):
|
| 12 |
prompt: str
|