WICKED4950 commited on
Commit
092bb03
·
verified ·
1 Parent(s): 7e3c3ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ def predict(user_input):
16
  inputs = tokenizer(user_input, return_tensors="tf", padding=True, truncation=True)
17
 
18
  # Generate the response using the model
19
- response_ids = model.generate(
20
  inputs['input_ids'],
21
  max_length=128, # Set max length of response
22
  do_sample=True, # Sampling for variability
 
16
  inputs = tokenizer(user_input, return_tensors="tf", padding=True, truncation=True)
17
 
18
  # Generate the response using the model
19
+ response_id = model.generate(
20
  inputs['input_ids'],
21
  max_length=128, # Set max length of response
22
  do_sample=True, # Sampling for variability