Lisibonny commited on
Commit
14d3abf
1 Parent(s): 079ed12

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -137,7 +137,7 @@ def main():
137
  text=remove_URL(text)
138
  inputs = tokenizer(query, text[0:512], return_tensors='tf')
139
  outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
140
- st.write(outputs)
141
  answer_start_index = int(tf.math.argmax(outputs.start_logits, axis=-1)[0])
142
  answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
143
  #answer_start_scores = tf.nn.softmax(outputs.start_logits)
 
137
  text=remove_URL(text)
138
  inputs = tokenizer(query, text[0:512], return_tensors='tf')
139
  outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
140
+ st.write("HI")
141
  answer_start_index = int(tf.math.argmax(outputs.start_logits, axis=-1)[0])
142
  answer_end_index = int(tf.math.argmax(outputs.end_logits, axis=-1)[0])
143
  #answer_start_scores = tf.nn.softmax(outputs.start_logits)