Spaces:
Runtime error
Runtime error
Sreyan Ghosh
commited on
Commit
·
b82d4c0
1
Parent(s):
5f171b8
changed temp
Browse files
app.py
CHANGED
@@ -213,7 +213,7 @@ def predict(filepath, question):
|
|
213 |
lang_x=prompt,
|
214 |
eos_token_id=tokenizer.eos_token_id,
|
215 |
max_new_tokens=256,
|
216 |
-
temperature=0.
|
217 |
|
218 |
output_decoded = tokenizer.decode(output).split(tokenizer.sep_token)[-1].replace(tokenizer.eos_token, '').replace(tokenizer.pad_token, '').replace('<|endofchunk|>', '')
|
219 |
|
|
|
213 |
lang_x=prompt,
|
214 |
eos_token_id=tokenizer.eos_token_id,
|
215 |
max_new_tokens=256,
|
216 |
+
temperature=0.0)[0]
|
217 |
|
218 |
output_decoded = tokenizer.decode(output).split(tokenizer.sep_token)[-1].replace(tokenizer.eos_token, '').replace(tokenizer.pad_token, '').replace('<|endofchunk|>', '')
|
219 |
|