Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
| 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
|