haramkoo commited on
Commit
3092d79
·
1 Parent(s): 4133bdb

added examples

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,15 +22,15 @@ examples = [
22
  @st.cache
23
  def genQuestion(model_choice, context):
24
  # global descriptions
25
- if model_choice=="interview-question-remake":
26
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-question-remake")
27
  tok = BartTokenizer.from_pretrained("hyechanjun/interview-question-remake")
28
  # descriptions = "Interview question remake is a model that..."
29
- elif model_choice=="interview-length-tagged":
30
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-length-tagged")
31
  tok = BartTokenizer.from_pretrained("hyechanjun/interview-length-tagged")
32
  # descriptions = "Interview question tagged is a model that..."
33
- elif model_choice=="reverse-interview-question":
34
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/reverse-interview-question")
35
  tok = BartTokenizer.from_pretrained("hyechanjun/reverse-interview-question")
36
  # descriptions = "Reverse interview question is a model that..."
 
22
  @st.cache
23
  def genQuestion(model_choice, context):
24
  # global descriptions
25
+ if model_choice=="Base model":
26
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-question-remake")
27
  tok = BartTokenizer.from_pretrained("hyechanjun/interview-question-remake")
28
  # descriptions = "Interview question remake is a model that..."
29
+ elif model_choice=="Lengthed model":
30
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-length-tagged")
31
  tok = BartTokenizer.from_pretrained("hyechanjun/interview-length-tagged")
32
  # descriptions = "Interview question tagged is a model that..."
33
+ elif model_choice=="Reversed model":
34
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/reverse-interview-question")
35
  tok = BartTokenizer.from_pretrained("hyechanjun/reverse-interview-question")
36
  # descriptions = "Reverse interview question is a model that..."