Abinaya Mahendiran commited on
Commit
6ec60b9
1 Parent(s): 940aa20

Updated app

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -5,7 +5,6 @@
5
  # Install necessary libraries
6
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
7
  import streamlit as st
8
- #from streamlit_tags import st_tags
9
  from pprint import pprint
10
  import json
11
 
@@ -35,4 +34,4 @@ st.markdown(
35
  )
36
  img = st.sidebar.image("images/tamil_logo.png", width=100)
37
  add_text_sidebar = st.sidebar.title("Select demo:")
38
- sampling_mode = st.sidebar.selectbox("select a Mode", index=0, options=["Top Sampling", "Beam Search"])
 
5
  # Install necessary libraries
6
  from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
7
  import streamlit as st
 
8
  from pprint import pprint
9
  import json
10
 
 
34
  )
35
  img = st.sidebar.image("images/tamil_logo.png", width=100)
36
  add_text_sidebar = st.sidebar.title("Select demo:")
37
+ sampling_mode = st.sidebar.selectbox("select a demo", index=0, options=["Text Generation", "Text Classification"])