Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -432,43 +432,9 @@ def sidebar():
|
|
| 432 |
format_func=lambda x: x.split(" ", 1)[1]
|
| 433 |
)
|
| 434 |
|
| 435 |
-
|
| 436 |
-
st.sidebar.subheader('Select Model')
|
| 437 |
-
models = ['3.5 Turbo', '4.0', 'Azure', 'Gemini', 'Ollama', 'Mistral'] # Example list of models
|
| 438 |
-
selected_model = st.sidebar.selectbox('Select Model', models)
|
| 439 |
-
|
| 440 |
-
api_key = st.sidebar.text_input('Enter API Key', type='password')
|
| 441 |
-
|
| 442 |
-
if selected_model == 'Azure':
|
| 443 |
-
endpoint = st.sidebar.text_input("Enter endpoint", type="url")
|
| 444 |
-
deployment_name = st.sidebar.text_input("Enter deployment name")
|
| 445 |
-
|
| 446 |
-
if st.sidebar.button('Run Model'):
|
| 447 |
-
st.session_state['api_key'] = api_key
|
| 448 |
-
|
| 449 |
-
if selected_model == '3.5 Turbo':
|
| 450 |
-
st.session_state['chat'] = ChatOpenAI(temperature=0, model="gpt-3.5-turbo", api_key=api_key)
|
| 451 |
-
elif selected_model == 'Azure':
|
| 452 |
-
st.session_state['chat'] = AzureChatOpenAI(
|
| 453 |
-
azure_endpoint=endpoint,
|
| 454 |
-
deployment_name=deployment_name,
|
| 455 |
-
openai_api_version="2023-03-15-preview",
|
| 456 |
-
api_key=api_key,
|
| 457 |
-
openai_api_type="Azure"
|
| 458 |
-
)
|
| 459 |
-
elif selected_model == '4.0':
|
| 460 |
-
st.session_state['chat'] = ChatOpenAI(temperature=0, model="gpt-4.0", api_key=api_key)
|
| 461 |
-
elif selected_model == 'Gemini':
|
| 462 |
st.session_state['chat'] = ChatGoogleGenerativeAI(model="gemini-1.5-pro",temperature=0,api_key=api_key,convert_system_message_to_human=True)
|
| 463 |
-
|
| 464 |
-
st.session_state['chat'] = ChatOpenAI(temperature=0, model="ollama-1", api_key=api_key)
|
| 465 |
-
elif selected_model == 'Mistral':
|
| 466 |
-
st.session_state['chat'] = ChatOpenAI(temperature=0, model="mistral-1", api_key=api_key)
|
| 467 |
-
else:
|
| 468 |
-
st.error('Invalid model selection')
|
| 469 |
-
return
|
| 470 |
-
|
| 471 |
-
st.success(f'Model {selected_model} is running...')
|
| 472 |
|
| 473 |
return selected_page.split(" ", 1)[1]
|
| 474 |
|
|
|
|
| 432 |
format_func=lambda x: x.split(" ", 1)[1]
|
| 433 |
)
|
| 434 |
|
| 435 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 436 |
st.session_state['chat'] = ChatGoogleGenerativeAI(model="gemini-1.5-pro",temperature=0,api_key=api_key,convert_system_message_to_human=True)
|
| 437 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 438 |
|
| 439 |
return selected_page.split(" ", 1)[1]
|
| 440 |
|