dcahn12 commited on
Commit
530df32
·
1 Parent(s): 1dbaf53

Fix import

Browse files
Files changed (1) hide show
  1. 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 serve.infer_utils import load_video_into_frames
22
- from serve.utils import load_image, image_ext, video_ext
23
- from serve.gradio_utils import Chat, tos_markdown, learn_more_markdown, title_markdown, block_css
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