Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,7 @@ tokenizer.pad_token = tokenizer.eos_token
|
|
16 |
with strategy.scope():
|
17 |
model = TFAutoModelForCausalLM.from_pretrained(name)
|
18 |
|
19 |
-
def raw_pred(input, model, tokenizer, max_length=
|
20 |
input_ids = tokenizer.encode(input, return_tensors='tf')
|
21 |
|
22 |
# Initialize variables
|
|
|
16 |
with strategy.scope():
|
17 |
model = TFAutoModelForCausalLM.from_pretrained(name)
|
18 |
|
19 |
+
def raw_pred(input, model, tokenizer, max_length=1024, temperature=0.2):
|
20 |
input_ids = tokenizer.encode(input, return_tensors='tf')
|
21 |
|
22 |
# Initialize variables
|