Aaravkumar commited on
Commit
d09b2ed
·
verified ·
1 Parent(s): 1de733d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -1,5 +1,6 @@
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
 
3
 
4
  from loader import Loader
5
  from chunker import Chunker
@@ -8,7 +9,7 @@ from vector import VectorStorage
8
  from retriever import Retriever
9
 
10
 
11
- client = InferenceClient("meta-llama/Llama-3.2-3B-Instruct")
12
 
13
  def process_document(file):
14
  """Systemic Entry Point: Converts PDF to Searchable Index"""
 
1
  import gradio as gr
2
  from huggingface_hub import InferenceClient
3
+ import os
4
 
5
  from loader import Loader
6
  from chunker import Chunker
 
9
  from retriever import Retriever
10
 
11
 
12
+ client = InferenceClient("mistralai/Mistral-7B-Instruct-v0.3", token=os.getenv("HF_TOKEN"))
13
 
14
  def process_document(file):
15
  """Systemic Entry Point: Converts PDF to Searchable Index"""