storresbusquets commited on
Commit
cad33a9
·
1 Parent(s): 6b12e1e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -210,9 +210,9 @@ class GradioInference:
210
 
211
  llm_model = "meta-llama/Llama-2-7b-chat-hf"
212
 
213
- llm_tokenizer = AutoTokenizer.from_pretrained(model)
214
 
215
- pipeline = transformers.pipeline(
216
  "text-generation", #task
217
  model=llm_model,
218
  tokenizer=llm_tokenizer,
@@ -225,7 +225,8 @@ class GradioInference:
225
  num_return_sequences=1,
226
  eos_token_id=tokenizer.eos_token_id
227
  )
228
-
 
229
  template = """
230
  Write a concise summary of the following text delimited by triple backquotes.
231
  ```{text}```
 
210
 
211
  llm_model = "meta-llama/Llama-2-7b-chat-hf"
212
 
213
+ llm_tokenizer = AutoTokenizer.from_pretrained(llm_model)
214
 
215
+ pipeline = pipeline(
216
  "text-generation", #task
217
  model=llm_model,
218
  tokenizer=llm_tokenizer,
 
225
  num_return_sequences=1,
226
  eos_token_id=tokenizer.eos_token_id
227
  )
228
+ llm = HuggingFacePipeline(pipeline = pipeline, model_kwargs = {'temperature':0})
229
+
230
  template = """
231
  Write a concise summary of the following text delimited by triple backquotes.
232
  ```{text}```