Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ import os
|
|
8 |
|
9 |
|
10 |
# Initialize the model pipeline
|
11 |
-
generator = pipeline('text-generation', model='Locutusque/ChatHercules-2.5-Mistral-7B-DPO', torch_dtype=torch.bfloat16, os.environ["HF"])
|
12 |
@spaces.GPU
|
13 |
def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_length):
|
14 |
# Generate text using the model
|
|
|
8 |
|
9 |
|
10 |
# Initialize the model pipeline
|
11 |
+
generator = pipeline('text-generation', model='Locutusque/ChatHercules-2.5-Mistral-7B-DPO', torch_dtype=torch.bfloat16, token=os.environ["HF"])
|
12 |
@spaces.GPU
|
13 |
def generate_text(prompt, temperature, top_p, top_k, repetition_penalty, max_length):
|
14 |
# Generate text using the model
|