Spaces:
Sleeping
Sleeping
Commit
·
997b5bb
1
Parent(s):
b6e1e0b
Update app.py
Browse files
app.py
CHANGED
@@ -55,7 +55,7 @@ def predict_text(text):
|
|
55 |
|
56 |
def generate_text(model, prompt, max_length=150, temperature=1.0):
|
57 |
input_ids = tok.encode(prompt, return_tensors='pt')
|
58 |
-
output =
|
59 |
input_ids=input_ids,
|
60 |
max_length=max_length + len(input_ids[0]),
|
61 |
temperature=temperature,
|
|
|
55 |
|
56 |
def generate_text(model, prompt, max_length=150, temperature=1.0):
|
57 |
input_ids = tok.encode(prompt, return_tensors='pt')
|
58 |
+
output = model_finetuned.generate(
|
59 |
input_ids=input_ids,
|
60 |
max_length=max_length + len(input_ids[0]),
|
61 |
temperature=temperature,
|