FuriouslyAsleep commited on
Commit
c1f5f3b
·
1 Parent(s): 79846b9

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -81,8 +81,9 @@ tokenizer = MarkupLMTokenizer(
81
 
82
  def greet(name):
83
 
84
- question = name
85
- single_html_string = "<h1>Grid Fin</h1><table><tr><td><p>Cost Risk</p></td><td><p>None</p></td></tr><tr><td><p>Schedule Risk</p></td><td><p>None</p></td></tr></table><h1>Propulsion</h1><table><tr><td><p>Cost Risk</p></td><td><p>Yes, material overrun</p></td></tr><tr><td><p>Schedule Risk</p></td><td><p>None</p></td></tr></table><h1>Nose Cone</h1><table><tr><td><p>Cost Risk</p></td><td><p>None</p></td></tr><tr><td><p>Schedule Risk</p></td><td><p>None</p></td></tr></table>"
 
86
  inputs = tokenizer.encode_plus(question, single_html_string, return_tensors="pt", padding="max_length", max_length=30, truncation=True)
87
 
88
  answer_start_scores, answer_end_scores = model(**inputs, return_dict=False)
 
81
 
82
  def greet(name):
83
 
84
+ question = name[1]
85
+ single_html_string = name[0]
86
+ #single_html_string = "<h1>Grid Fin</h1><table><tr><td><p>Cost Risk</p></td><td><p>None</p></td></tr><tr><td><p>Schedule Risk</p></td><td><p>None</p></td></tr></table><h1>Propulsion</h1><table><tr><td><p>Cost Risk</p></td><td><p>Yes, material overrun</p></td></tr><tr><td><p>Schedule Risk</p></td><td><p>None</p></td></tr></table><h1>Nose Cone</h1><table><tr><td><p>Cost Risk</p></td><td><p>None</p></td></tr><tr><td><p>Schedule Risk</p></td><td><p>None</p></td></tr></table>"
87
  inputs = tokenizer.encode_plus(question, single_html_string, return_tensors="pt", padding="max_length", max_length=30, truncation=True)
88
 
89
  answer_start_scores, answer_end_scores = model(**inputs, return_dict=False)