Locutusque commited on
Commit
bd4be89
1 Parent(s): 01740bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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