MadhuP commited on
Commit
61c61b5
·
1 Parent(s): 0f926c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -7
app.py CHANGED
@@ -125,9 +125,7 @@ def predict_2(txt_url, normalized_content_with_style):
125
  confidence_scores_content,
126
  new_data,
127
  ) = (None, None, None, None, None)
128
-
129
  predicted_label_text, confidence_score_text = None, None
130
-
131
  if txt_url.startswith("http://") or txt_url.startswith("https://"):
132
  (
133
  predicted_label_title,
@@ -147,7 +145,6 @@ def predict_2(txt_url, normalized_content_with_style):
147
  max_length=512,
148
  return_tensors="pt",
149
  )
150
-
151
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
152
  test_input_ids = test_encodings["input_ids"].to(device)
153
  test_attention_mask = test_encodings["attention_mask"].to(device)
@@ -172,7 +169,6 @@ def predict_2(txt_url, normalized_content_with_style):
172
  confidence_score_text,
173
  #new,
174
  )
175
-
176
  def word_by_word(txt_url, normalized_content_with_style):
177
  if txt_url.startswith("http://") or txt_url.startswith("https://") or txt_url.startswith(""):
178
  (
@@ -235,8 +231,6 @@ def word_by_word(txt_url, normalized_content_with_style):
235
  confidence_score_text,
236
  new_word,
237
  )
238
-
239
-
240
  demo = gr.Interface(
241
  fn=word_by_word,
242
  inputs=[
@@ -254,5 +248,4 @@ demo = gr.Interface(
254
  gr.outputs.Textbox(label="word-by-word").style(show_copy_button=True),
255
  ],
256
  )
257
-
258
  demo.launch()
 
125
  confidence_scores_content,
126
  new_data,
127
  ) = (None, None, None, None, None)
 
128
  predicted_label_text, confidence_score_text = None, None
 
129
  if txt_url.startswith("http://") or txt_url.startswith("https://"):
130
  (
131
  predicted_label_title,
 
145
  max_length=512,
146
  return_tensors="pt",
147
  )
 
148
  device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
149
  test_input_ids = test_encodings["input_ids"].to(device)
150
  test_attention_mask = test_encodings["attention_mask"].to(device)
 
169
  confidence_score_text,
170
  #new,
171
  )
 
172
  def word_by_word(txt_url, normalized_content_with_style):
173
  if txt_url.startswith("http://") or txt_url.startswith("https://") or txt_url.startswith(""):
174
  (
 
231
  confidence_score_text,
232
  new_word,
233
  )
 
 
234
  demo = gr.Interface(
235
  fn=word_by_word,
236
  inputs=[
 
248
  gr.outputs.Textbox(label="word-by-word").style(show_copy_button=True),
249
  ],
250
  )
 
251
  demo.launch()