Carlos Salgado commited on
Commit
5ee94a9
·
1 Parent(s): 4da192d

fix MODEL_NAME constant

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. scripts.py +1 -2
app.py CHANGED
@@ -3,7 +3,7 @@ import os
3
  import streamlit as st
4
  import tempfile
5
 
6
- from scripts import generate_metadata, ingest, model_name
7
 
8
 
9
  st.title('DocVerifyRAG')
 
3
  import streamlit as st
4
  import tempfile
5
 
6
+ from scripts import generate_metadata, ingest, MODEL_NAME
7
 
8
 
9
  st.title('DocVerifyRAG')
scripts.py CHANGED
@@ -12,8 +12,8 @@ from langchain_text_splitters import RecursiveCharacterTextSplitter
12
 
13
  load_dotenv()
14
 
 
15
 
16
- import io
17
 
18
  def ingest(file_path):
19
  extension = os.path.splitext(file_path)[1].lower()
@@ -72,7 +72,6 @@ def generate_metadata(docs):
72
  )
73
 
74
  # Call the LLM with the JSON schema
75
- model_name = "mistralai/Mixtral-8x7B-Instruct-v0.1"
76
  chat_completion = client.chat.completions.create(
77
  model=model_name,
78
  messages=[
 
12
 
13
  load_dotenv()
14
 
15
+ MODEL_NAME = "mistralai/Mixtral-8x7B-Instruct-v0.1"
16
 
 
17
 
18
  def ingest(file_path):
19
  extension = os.path.splitext(file_path)[1].lower()
 
72
  )
73
 
74
  # Call the LLM with the JSON schema
 
75
  chat_completion = client.chat.completions.create(
76
  model=model_name,
77
  messages=[