Update app.py

app.py

@@ -31,12 +31,14 @@ with col1:
 with col2:
     select_model = st.radio(
         "Select the model to use:",
-        ('gpt-ya', 'gpt-ya-1-1'), index = 1)
+        ('gpt-ya', 'gpt-ya-1-1', 'gpt-ya-1-1-160M'), index = 2)
 
     if select_model == 'gpt-ya':
         model = 'breadlicker45/gpt-ya'
     elif select_model == 'gpt-ya-1-1':
         model = 'BreadAi/gpt-YA-1-1_70M'
+    elif select_model == 'gpt-ya-1-1-160M':
+        model = 'BreadAi/gpt-YA-1-1_160M'
 
     with st.spinner('Loading Model... (This may take a while)'):
         generator = get_model()