Pijush2023 commited on
Commit
da91251
·
verified ·
1 Parent(s): 335ee48

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -126,12 +126,12 @@ gpt4o_mini_model = initialize_gpt4o_mini_model()
126
 
127
  # Existing embeddings and vector store for GPT-4o
128
  gpt_embeddings = OpenAIEmbeddings(api_key=os.environ['OPENAI_API_KEY'])
129
- gpt_vectorstore = PineconeVectorStore(index_name="italyv109102024", embedding=embeddings)
130
  gpt_retriever = gpt_vectorstore.as_retriever(search_kwargs={'k': 5})
131
 
132
  # New vector store setup for Phi-3.5
133
  phi_embeddings = embeddings
134
- phi_vectorstore = PineconeVectorStore(index_name="italyv109102024", embedding=embeddings)
135
  phi_retriever = phi_vectorstore.as_retriever(search_kwargs={'k': 5})
136
 
137
 
@@ -142,7 +142,7 @@ phi_retriever = phi_vectorstore.as_retriever(search_kwargs={'k': 5})
142
  from pinecone import Pinecone
143
  pc = Pinecone(api_key=os.environ['PINECONE_API_KEY'])
144
 
145
- index_name = "italyv109102024"
146
  vectorstore = PineconeVectorStore(index_name=index_name, embedding=embeddings)
147
  retriever = vectorstore.as_retriever(search_kwargs={'k': 5})
148
 
@@ -1571,7 +1571,7 @@ with gr.Blocks(theme='gradio/soft') as demo:
1571
  chatbot = gr.Chatbot([], elem_id="RADAR:Channel 94.1", bubble_full_width=False)
1572
  choice = gr.Radio(label="Select Style", choices=["Details", "Conversational"], value="Conversational",interactive=False,visible=False)
1573
  retrieval_mode = gr.Radio(label="Retrieval Mode", choices=["VDB", "KGF"], value="VDB",interactive=False,visible=False)
1574
- model_choice = gr.Dropdown(label="Choose Model", choices=["LM-2"], value="LM-2")
1575
 
1576
  # Link the dropdown change to handle_model_choice_change
1577
  model_choice.change(fn=handle_model_choice_change, inputs=model_choice, outputs=[retrieval_mode, choice, choice])
 
126
 
127
  # Existing embeddings and vector store for GPT-4o
128
  gpt_embeddings = OpenAIEmbeddings(api_key=os.environ['OPENAI_API_KEY'])
129
+ gpt_vectorstore = PineconeVectorStore(index_name="italyopenai", embedding=gpt_embeddings)
130
  gpt_retriever = gpt_vectorstore.as_retriever(search_kwargs={'k': 5})
131
 
132
  # New vector store setup for Phi-3.5
133
  phi_embeddings = embeddings
134
+ phi_vectorstore = PineconeVectorStore(index_name="italyopenai", embedding=embeddings)
135
  phi_retriever = phi_vectorstore.as_retriever(search_kwargs={'k': 5})
136
 
137
 
 
142
  from pinecone import Pinecone
143
  pc = Pinecone(api_key=os.environ['PINECONE_API_KEY'])
144
 
145
+ index_name = "italyopenai"
146
  vectorstore = PineconeVectorStore(index_name=index_name, embedding=embeddings)
147
  retriever = vectorstore.as_retriever(search_kwargs={'k': 5})
148
 
 
1571
  chatbot = gr.Chatbot([], elem_id="RADAR:Channel 94.1", bubble_full_width=False)
1572
  choice = gr.Radio(label="Select Style", choices=["Details", "Conversational"], value="Conversational",interactive=False,visible=False)
1573
  retrieval_mode = gr.Radio(label="Retrieval Mode", choices=["VDB", "KGF"], value="VDB",interactive=False,visible=False)
1574
+ model_choice = gr.Dropdown(label="Choose Model", choices=["LM-1"], value="LM-1")
1575
 
1576
  # Link the dropdown change to handle_model_choice_change
1577
  model_choice.change(fn=handle_model_choice_change, inputs=model_choice, outputs=[retrieval_mode, choice, choice])