Update app.py

app.py

@@ -12,13 +12,14 @@ all_input = st.text_area('模型输入', value="""
 top_p = st.slider('top_p', 0.0, 1.0, 0.95)
 temperature = st.slider('temperature', 0.0, 1.0, 0.85)
 max_tokens = st.slider('max tokens', 4, 512, 64)
+model_type = st.selectbox('model', ('medium', 'large', 'xl'))
 
 
 def completion(prompt):
     start = time.monotonic()
     resp = requests.post('https://welm.weixin.qq.com/v1/completions', json={
         'prompt': prompt,
-        'model': 'large',
+        'model': model_type,
         'max_tokens': max_tokens,
         'temperature': temperature,
         'top_p': top_p,