khoatran94 commited on
Commit
d7b6a67
·
1 Parent(s): 021d971

test cv extraction

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -97,6 +97,7 @@ def LLM_Inference(cv_text):
97
 
98
  Do not explain, comment or make up any more information that is not relative to the list of Information extraction. Respond in Vietnamese. Let's work this out in a step by step way to ensure the correct answer. [END].
99
  '''
 
100
  inputs = tokenizer(text, return_tensors='pt').to(device)
101
  with torch.no_grad():
102
  outputs = model.generate(
 
97
 
98
  Do not explain, comment or make up any more information that is not relative to the list of Information extraction. Respond in Vietnamese. Let's work this out in a step by step way to ensure the correct answer. [END].
99
  '''
100
+ text = 'short answer on who Lê Duẩn is'
101
  inputs = tokenizer(text, return_tensors='pt').to(device)
102
  with torch.no_grad():
103
  outputs = model.generate(