Vladislawoo commited on
Commit
997b5bb
·
1 Parent(s): b6e1e0b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = sber.generate(
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,