Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -27,7 +27,7 @@ def generate_response(model, tokenizer, prompt):
|
|
| 27 |
response = tokenizer.decode(output[0], skip_special_tokens=True)
|
| 28 |
print("Generated response::", response )
|
| 29 |
print("len(prompt)::", len(prompt) )
|
| 30 |
-
response = response[len(prompt) +
|
| 31 |
return response
|
| 32 |
|
| 33 |
def predict(input):
|
|
|
|
| 27 |
response = tokenizer.decode(output[0], skip_special_tokens=True)
|
| 28 |
print("Generated response::", response )
|
| 29 |
print("len(prompt)::", len(prompt) )
|
| 30 |
+
response = response[len(prompt) + 9:]
|
| 31 |
return response
|
| 32 |
|
| 33 |
def predict(input):
|