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 |
|