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