samiee2213 commited on
Commit
5f1b3b4
1 Parent(s): df3c808

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -36
app.py CHANGED
@@ -432,43 +432,9 @@ def sidebar():
432
  format_func=lambda x: x.split(" ", 1)[1]
433
  )
434
 
435
- # API Key and Model Selection
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
- elif selected_model == 'Ollama':
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