Mohammed-Altaf commited on
Commit
6b96a20
1 Parent(s): 04ae5c1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -28,10 +28,10 @@ def get_clean_response(response):
28
 
29
 
30
  def generate_text(input_text):
31
- input_ids = tokenizer.encode(input_text, return_tensors="pt")
32
 
33
  output = model.generate(
34
- input_ids,
35
  max_length=100,
36
  )
37
 
 
28
 
29
 
30
  def generate_text(input_text):
31
+ input_ids = tokenizer(input_text, return_tensors="pt")
32
 
33
  output = model.generate(
34
+ **input_ids,
35
  max_length=100,
36
  )
37