Robin246 commited on
Commit
7a6fcd6
·
verified ·
1 Parent(s): 756023e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -22,8 +22,8 @@ def generate_response(input_prompt, model_path):
22
  return generated_output, model_name
23
 
24
  def main():
25
- st.title("INXAI LLM Model")
26
- model_selection = st.selectbox("Choose a model", ["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)
 
22
  return generated_output, model_name
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)