Spaces:
Build error
Build error
added examples
Browse files
app.py
CHANGED
@@ -22,15 +22,15 @@ examples = [
|
|
22 |
@st.cache
|
23 |
def genQuestion(model_choice, context):
|
24 |
# global descriptions
|
25 |
-
if model_choice=="
|
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=="
|
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=="
|
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..."
|