ajs2440 commited on
Commit
e7eb773
·
2 Parent(s): a01a63e ab46e65

adding dropdowns

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -2,11 +2,17 @@ import gradio as gr
2
  import torch
3
  from transformers import BartForConditionalGeneration, BartTokenizer
4
 
 
5
  # initialize model + tok variables
6
  model = None
7
  tok = None
8
 
9
  # pass in Strings of model choice and input text for context
 
 
 
 
 
10
  def genQuestion(model_choice, context):
11
  if model_choice=="interview-question-remake":
12
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-question-remake")
 
2
  import torch
3
  from transformers import BartForConditionalGeneration, BartTokenizer
4
 
5
+ <<<<<<< HEAD
6
  # initialize model + tok variables
7
  model = None
8
  tok = None
9
 
10
  # pass in Strings of model choice and input text for context
11
+ =======
12
+ model = None
13
+ tok = None
14
+
15
+ >>>>>>> ajs
16
  def genQuestion(model_choice, context):
17
  if model_choice=="interview-question-remake":
18
  model = BartForConditionalGeneration.from_pretrained("hyechanjun/interview-question-remake")