Mary12 commited on
Commit
6a25e07
·
1 Parent(s): c8dd179

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -20
app.py CHANGED
@@ -58,50 +58,45 @@ def model(model_name):
58
  model_name = "timpal0l/mdeberta-v3-base-squad2"
59
  pipe = model(model_name)
60
 
61
- def qa_result(context, question, file, pipe = pipe):
62
  if file is not None:
63
  if len(question) == 0:
64
- text = "Ես չեմ կարողանա քեզ օգնել եթե ինձ չտաս հարցդ"
65
  else:
66
  allowed_types = [".pdf", ".txt", ".docx"]
67
  extension = "." + file.name.split(".")[-1].lower()
68
  if not extension in allowed_types:
69
- text = "Խնդրում եմ ներբեռնել .pdf, .txt, կամ .docx տիպի ֆայլեր։"
70
  else:
71
  if extension == allowed_types[0]:
72
  context = extract_text_from_pdf(file.name)
73
  elif extension == allowed_types[1]:
74
  context = extract_text_from_txt(file.name)
75
  else:
76
- context = extract_text_from_pdf(file.name)
77
- result = pipe(question=question, context=context)
78
- answered = result['answer']
79
- text = remove_references(answered)
80
-
81
- # text = remove_references(answered)
82
-
83
- else:
84
- if file is None and len(context) == 0 and len(question) == 0:
85
- text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն կամ ֆայլն ու հարցերը։"
86
  else:
87
  if len(context) == 0 and len(question) == 0:
88
- text = "Որպեսզի ես կարողանամ քեզ օգնել, ինձ պիտի տրամադրես համապատասխան տեքստն ու հարցերը։"
89
- elif len(context) == 0:
90
- text = "Ես չեմ կարողանա քեզ օգնել եթե ինձ չտրամադրես տեքստը"
91
  elif len(context) == 0:
92
- text = "Ես չեմ կարողանա քեզ օգնել եթե ինձ չտաս հարցդ"
 
 
93
  else:
94
  result = pipe(question=question, context=context)
95
  answered = result['answer']
96
  text = remove_references(answered)
97
-
98
  text = text.replace('(', '', 1)
99
- text = text.replace(',', '', len(text)-1)
100
-
101
  return text.capitalize()
102
 
103
 
104
 
 
105
  theme = gr.themes.Soft().set(
106
  body_background_fill='*background_fill_secondary',
107
  body_text_color_subdued='*body_text_color',
 
58
  model_name = "timpal0l/mdeberta-v3-base-squad2"
59
  pipe = model(model_name)
60
 
61
+ def qa_result(context, question, file, pipe=pipe):
62
  if file is not None:
63
  if len(question) == 0:
64
+ text = "Ես չեմ կարողանալ քեզ օգնել եթե ինձ չտաս հարցը"
65
  else:
66
  allowed_types = [".pdf", ".txt", ".docx"]
67
  extension = "." + file.name.split(".")[-1].lower()
68
  if not extension in allowed_types:
69
+ text = "Խնդրում եմ ներբեռնել .pdf, .txt, կամ .docx ֆայլեր"
70
  else:
71
  if extension == allowed_types[0]:
72
  context = extract_text_from_pdf(file.name)
73
  elif extension == allowed_types[1]:
74
  context = extract_text_from_txt(file.name)
75
  else:
76
+ context = extract_text_from_doc(file.name)
77
+ result = pipe(question=question, context=context)
78
+ answered = result['answer']
79
+ text = remove_references(answered)
 
 
 
 
 
 
80
  else:
81
  if len(context) == 0 and len(question) == 0:
82
+ text = "Որպեսզի ես կարողանամ քեզ օգնել, դուք պետք է տրամադրեք տեքստ և հարցեր"
 
 
83
  elif len(context) == 0:
84
+ text = "Ես չեմ կարողանամ քեզ օգնել եթե դուք չտարածեք տեքստը"
85
+ elif len(question) == 0:
86
+ text = "Ես չեմ կարողանամ քեզ օգնել եթե դուք չտասնեք հարցը"
87
  else:
88
  result = pipe(question=question, context=context)
89
  answered = result['answer']
90
  text = remove_references(answered)
91
+
92
  text = text.replace('(', '', 1)
93
+ text = text.rstrip(',')
94
+
95
  return text.capitalize()
96
 
97
 
98
 
99
+
100
  theme = gr.themes.Soft().set(
101
  body_background_fill='*background_fill_secondary',
102
  body_text_color_subdued='*body_text_color',