Spaces:
Runtime error
Runtime error
Upload app.py
Browse files
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 |
|