Spaces:
Runtime error
Runtime error
storresbusquets
commited on
Commit
·
cad33a9
1
Parent(s):
6b12e1e
Update app.py
Browse files
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(
|
214 |
|
215 |
-
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}```
|