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 |
|