Update app.py
Browse files
app.py
CHANGED
|
@@ -94,7 +94,7 @@ def chat(video_number,token):
|
|
| 94 |
output = model.generate(**inputs, **generate_kwargs)
|
| 95 |
generated_text = processor.batch_decode(output, skip_special_tokens=True)
|
| 96 |
|
| 97 |
-
return generated_text
|
| 98 |
|
| 99 |
demo = gr.Interface(
|
| 100 |
fn=chat,
|
|
|
|
| 94 |
output = model.generate(**inputs, **generate_kwargs)
|
| 95 |
generated_text = processor.batch_decode(output, skip_special_tokens=True)
|
| 96 |
|
| 97 |
+
return generated_text[0][45:]
|
| 98 |
|
| 99 |
demo = gr.Interface(
|
| 100 |
fn=chat,
|