Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -202,14 +202,15 @@ def generate_video(model_name: str, text: str, video_path: str,
|
|
202 |
buffer = ""
|
203 |
for new_text in streamer:
|
204 |
buffer += new_text
|
|
|
205 |
time.sleep(0.01)
|
206 |
yield buffer
|
207 |
|
208 |
# Define examples for image and video inference
|
209 |
image_examples = [
|
210 |
-
["
|
211 |
-
["
|
212 |
-
["
|
213 |
]
|
214 |
|
215 |
video_examples = [
|
|
|
202 |
buffer = ""
|
203 |
for new_text in streamer:
|
204 |
buffer += new_text
|
205 |
+
buffer = buffer.replace("<|im_end|>", "")
|
206 |
time.sleep(0.01)
|
207 |
yield buffer
|
208 |
|
209 |
# Define examples for image and video inference
|
210 |
image_examples = [
|
211 |
+
["ocr the image", "example/image1.png"],
|
212 |
+
["explain the scene", "example/image2.jpg"],
|
213 |
+
["fill the correct numbers", "example/image3.png"],
|
214 |
]
|
215 |
|
216 |
video_examples = [
|