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..."
|