Robin246 commited on
Commit
3dbf704
1 Parent(s): 7a6fcd6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -23,7 +23,7 @@ def generate_response(input_prompt, model_path):
23
 
24
  def main():
25
  st.title("INXAI LLM Model\nCompare with base models")
26
- model_selection = st.selectbox("Choose INXAI from the dropdown", ["google/flan-t5-small", "MBZUAI/LaMini-Flan-T5-77M", "Robin246/inxai"])
27
  input_prompt = st.text_input("Enter input text")
28
  if st.button("Generate"):
29
  reply, model_name = generate_response(input_prompt, model_selection)
 
23
 
24
  def main():
25
  st.title("INXAI LLM Model\nCompare with base models")
26
+ model_selection = st.selectbox("Choose INXAI from the dropdown", ["google/flan-t5-small", "MBZUAI/LaMini-Flan-T5-77M", "Robin246/inxai_v1.1"])
27
  input_prompt = st.text_input("Enter input text")
28
  if st.button("Generate"):
29
  reply, model_name = generate_response(input_prompt, model_selection)