Update app.py

app.py

@@ -9,12 +9,12 @@ os.environ["PINECONE_API_KEY"] = st.secrets["PINECONE_API_KEY"]
 # Sidebar for selecting the chatbot
 selected_chatbot = st.sidebar.radio("Select Chatbot", ("OpenAI", "Llama 2"))
 if selected_chatbot == "OpenAI":
-    from src import openai_call
+    from openai_call import openai_call
 elif selected_chatbot == "Llama 2":
     st.warning(
         "It might take some time to get response becuase of the size of Llama 2 model ⚠️"
     )
-    from src import llama_call
+    from llama_call import llama_call
 
 # Initialize chat history
 if "messages" not in st.session_state: