Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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
|
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)
|