anand004 commited on
Commit
c0f02eb
1 Parent(s): 0b0135c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -13,6 +13,7 @@ import torch
13
  from PIL import Image
14
  from chromadb.utils import embedding_functions
15
  from chromadb.utils.data_loaders import ImageLoader
 
16
  from langchain import PromptTemplate
17
  from langchain.text_splitter import RecursiveCharacterTextSplitter
18
  from langchain_community.llms import HuggingFaceEndpoint
@@ -20,6 +21,7 @@ from pdfminer.high_level import extract_text
20
  from transformers import LlavaNextForConditionalGeneration, LlavaNextProcessor
21
  from utils import *
22
 
 
23
  if torch.cuda.is_available():
24
  processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf")
25
  vision_model = LlavaNextForConditionalGeneration.from_pretrained(
 
13
  from PIL import Image
14
  from chromadb.utils import embedding_functions
15
  from chromadb.utils.data_loaders import ImageLoader
16
+ from gradio.themes.utils import sizes
17
  from langchain import PromptTemplate
18
  from langchain.text_splitter import RecursiveCharacterTextSplitter
19
  from langchain_community.llms import HuggingFaceEndpoint
 
21
  from transformers import LlavaNextForConditionalGeneration, LlavaNextProcessor
22
  from utils import *
23
 
24
+
25
  if torch.cuda.is_available():
26
  processor = LlavaNextProcessor.from_pretrained("llava-hf/llava-v1.6-mistral-7b-hf")
27
  vision_model = LlavaNextForConditionalGeneration.from_pretrained(