Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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
|
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 =
|
77 |
-
|
78 |
-
|
79 |
-
|
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.
|
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',
|