amir22010 commited on
Commit
c8e5dba
·
1 Parent(s): 8df7942

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -55,6 +55,7 @@ def perform_asde_inference(text, dataset, model_id):
55
  tokenized_text = tokenizer_english(bos_instruction + text + delim_instruct + eos_instruct, return_tensors="pt")
56
  output = double_english_generator.generate(tokenized_text.input_ids,max_length=512)
57
  model_generated = tokenizer_english.decode(output[0], skip_special_tokens=True)
 
58
  '''
59
  elif model_id == "PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model":
60
  tokenized_text = tokenizer_multilingual(bos_instruction + text + delim_instruct + eos_instruct, return_tensors="pt")
@@ -73,7 +74,7 @@ def perform_asde_inference(text, dataset, model_id):
73
  if not text:
74
  true_doubles = pd.DataFrame(list(map(list, zip(ast.literal_eval(true_aspect), ast.literal_eval(true_sentiment)))),columns=['Aspect','Sentiment'])
75
  else:
76
- true_doubles = pd.DataFrame()
77
 
78
  return pred_doubles, true_doubles, text, model_generated
79
 
 
55
  tokenized_text = tokenizer_english(bos_instruction + text + delim_instruct + eos_instruct, return_tensors="pt")
56
  output = double_english_generator.generate(tokenized_text.input_ids,max_length=512)
57
  model_generated = tokenizer_english.decode(output[0], skip_special_tokens=True)
58
+
59
  '''
60
  elif model_id == "PyABSA_Hospital_Multilingual_allenai/tk-instruct-base-def-pos_FinedTuned_Model":
61
  tokenized_text = tokenizer_multilingual(bos_instruction + text + delim_instruct + eos_instruct, return_tensors="pt")
 
74
  if not text:
75
  true_doubles = pd.DataFrame(list(map(list, zip(ast.literal_eval(true_aspect), ast.literal_eval(true_sentiment)))),columns=['Aspect','Sentiment'])
76
  else:
77
+ true_doubles = pd.DataFrame([["",""]],columns=['Aspect','Sentiment'])
78
 
79
  return pred_doubles, true_doubles, text, model_generated
80