breadlicker45 commited on
Commit
ccfbfd7
·
1 Parent(s): 831b959

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -31,14 +31,14 @@ with col1:
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
- ('gpt-ya', 'gpt-ya-1-1', 'gpt-ya-1-1-160M'), index = 2)
35
 
36
- if select_model == 'gpt-ya':
37
- model = 'breadlicker45/gpt-ya'
38
- elif select_model == 'gpt-ya-1-1':
39
- model = 'BreadAi/gpt-YA-1-1_70M'
40
- elif select_model == 'gpt-ya-1-1-160M':
41
- model = 'BreadAi/gpt-YA-1-1_160M'
42
 
43
  with st.spinner('Loading Model... (This may take a while)'):
44
  generator = get_model()
 
31
  with col2:
32
  select_model = st.radio(
33
  "Select the model to use:",
34
+ ('gpt-youtube'), index = 0)
35
 
36
+ if select_model == 'gpt-youtube':
37
+ model = 'BreadAi/gpt-Youtube'
38
+ elif select_model == 'gpt-null':
39
+ model = 'BreadAi/gpt-Youtube'
40
+ elif select_model == 'gpt-null1':
41
+ model = 'BreadAi/gpt-Youtube'
42
 
43
  with st.spinner('Loading Model... (This may take a while)'):
44
  generator = get_model()