Update app.py

app.py

@@ -4,6 +4,10 @@ import time
 
 def infer(prompt, model_name, max_new_tokens=10, temperature=0.0, top_p=1.0):
 
+    model_name_map = {
+        "GPT-JT-6B-v1": "Together-gpt-JT-6B-v1",
+    }
+
     my_post_dict = {
         "type": "general",
         "payload": {
@@ -11,7 +15,7 @@ def infer(prompt, model_name, max_new_tokens=10, temperature=0.0, top_p=1.0):
             "n": 1,
             "temperature": float(temperature),
             "top_p": float(top_p),
-            "model": model_name,
+            "model": model_name_map[model_name],
             "prompt": [prompt],
             "request_type": "language-model-inference",
             "stop": None,
@@ -54,7 +58,7 @@ generated_area.markdown("(Generate here)")
 
 button_submit = st.button("Submit")
    
-model_name = st.selectbox("Model", ["Together-gpt-J-6B-ProxAdam-50x"])
+model_name = st.selectbox("Model", ["GPT-JT-6B-v1"])
 max_new_tokens = st.text_input('Max new tokens', "10")
 temperature = st.text_input('temperature', "0.0")
 top_p = st.text_input('top_p', "1.0")