Update app.py
Browse files
app.py
CHANGED
@@ -118,7 +118,7 @@ def imageMode(image, question):
|
|
118 |
pad_token_id=tokenizer_text.pad_token_id
|
119 |
)
|
120 |
)
|
121 |
-
text_pred = prediction[0].
|
122 |
return text_pred
|
123 |
|
124 |
def audioMode(audio):
|
|
|
118 |
pad_token_id=tokenizer_text.pad_token_id
|
119 |
)
|
120 |
)
|
121 |
+
text_pred = prediction[0].strip('<|endoftext|>').rstrip("\n")
|
122 |
return text_pred
|
123 |
|
124 |
def audioMode(audio):
|