Spaces:
Paused
Paused
dcahn12
commited on
Commit
·
530df32
1
Parent(s):
1dbaf53
Fix import
Browse files- gradio_web_server.py +3 -5
gradio_web_server.py
CHANGED
@@ -5,7 +5,6 @@ import torch
|
|
5 |
import gradio as gr
|
6 |
from fastapi import FastAPI
|
7 |
import os
|
8 |
-
# os.environ['CUDA_VISIBLE_DEVICES'] = '0'
|
9 |
from PIL import Image
|
10 |
import tempfile
|
11 |
from decord import VideoReader, cpu
|
@@ -13,14 +12,13 @@ from transformers import TextStreamer
|
|
13 |
import argparse
|
14 |
|
15 |
import sys
|
16 |
-
sys.path.insert(0, os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "Evaluation"))
|
17 |
from llava.constants import DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
|
18 |
from llava.conversation import conv_templates, SeparatorStyle, Conversation
|
19 |
from llava.mm_utils import process_images
|
20 |
|
21 |
-
from
|
22 |
-
from
|
23 |
-
from
|
24 |
|
25 |
|
26 |
|
|
|
5 |
import gradio as gr
|
6 |
from fastapi import FastAPI
|
7 |
import os
|
|
|
8 |
from PIL import Image
|
9 |
import tempfile
|
10 |
from decord import VideoReader, cpu
|
|
|
12 |
import argparse
|
13 |
|
14 |
import sys
|
|
|
15 |
from llava.constants import DEFAULT_IMAGE_TOKEN, DEFAULT_IM_START_TOKEN, DEFAULT_IM_END_TOKEN
|
16 |
from llava.conversation import conv_templates, SeparatorStyle, Conversation
|
17 |
from llava.mm_utils import process_images
|
18 |
|
19 |
+
from .infer_utils import load_video_into_frames
|
20 |
+
from .utils import load_image, image_ext, video_ext
|
21 |
+
from .gradio_utils import Chat, tos_markdown, learn_more_markdown, title_markdown, block_css
|
22 |
|
23 |
|
24 |
|