length tags

app.py

@@ -21,7 +21,7 @@ examples = [
 
 # pass in Strings of model choice and input text for context
 @st.cache
-def genQuestion(model_choice, context):
+def genQuestion(model_choice, context, tag):
     # global descriptions
     if model_choice=="Base model":
         model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-question-remake")
@@ -30,6 +30,26 @@ def genQuestion(model_choice, context):
     elif model_choice=="Lengthed model":
         model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-length-tagged")
         tok = BartTokenizer.from_pretrained("hyechanjun/interview-length-tagged")
+        if (tag == '1-10'):
+            context += ' <TEN>'
+        elif (tag == '11-20'):
+            context += ' <TWENTY>'
+        elif (tag == '21-30'):
+            context += ' <THIRTY>'
+        elif (tag == '31-40'):
+            context += ' <FORTY>'
+        elif (tag == '51-60'):
+            context += ' <FIFTY>'
+        elif (tag == '61-70'):
+            context += ' <SIXTY>'
+        elif (tag == '71-80'):
+            context += ' <SEVENTY>'
+        elif (tag == '81-90'):
+            context += ' <EIGHTY>'
+        elif (tag == '81-90'):
+            context += ' <NINETY>'
+        elif (tag == '91+'):
+            context += ' <HUNDRED>'
         # descriptions = "Interview question tagged is a model that..."
     elif model_choice=="Reverse model":
         model = BartForConditionalGeneration.from_pretrained("hyechanjun/reverse-interview-question")
@@ -56,9 +76,9 @@ st.caption("With the advent of machine learning, it has become increasingly clea
 if 'button_sent' not in st.session_state:
     st.session_state.button_sent = False
 
-maxl, minl = st.columns(2)
+col1, col2,  = st.columns(3)
 
-context_option = minl.selectbox(
+context_option = col2.selectbox(
                 'Feel free to choose one of our premade contexts',
                 ('Select one','Elon Musk', 'Fashion designer', 'Young entrepreneur', 'Michael Jordan')
 )
@@ -74,7 +94,7 @@ elif context_option == 'Young entrepreneur':
 else:
     context_example = examples[3]
 
-option = maxl.selectbox(
+option = col1.selectbox(
                 'Please select a model.',
                 ('Base model', 'Lengthed model', 'Reverse model'))
 
@@ -87,13 +107,18 @@ elif option == 'Lengthed model':
 elif option == 'Reverse model':
     st.write("This model asks a question that would have resulted in the context you provide (a.k.a. it traverses backward through the interview)")
 
+if option == 'Lengthed model':
+    context_length = col3.selectbox('Length of response'
+                                    ('1-10', '11-20', '21-30', '31-40', '41-50', '51-60', '61-70', '71-80', '81-90', '91+'))
+
+
 # Input fields
 input = st.text_area('Context', value=context_example)                                    # user inputs context to construct a response (str)
 
 
 if st.button('Submit') or st.session_state.button_sent:
     with st.spinner('Generating a response...'):
-        output = genQuestion(option, input)
+        output = genQuestion(option, input, tag)
         print(output)
     # st.write(output)
     st.session_state.button_sent = True