samariddin commited on
Commit
fbfd6f7
1 Parent(s): 4778387
.idea/.gitignore ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ # Default ignored files
2
+ /shelf/
3
+ /workspace.xml
.idea/Uz-Text-Summarization.iml ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ <?xml version="1.0" encoding="UTF-8"?>
2
+ <module type="PYTHON_MODULE" version="4">
3
+ <component name="NewModuleRootManager">
4
+ <content url="file://$MODULE_DIR$" />
5
+ <orderEntry type="inheritedJdk" />
6
+ <orderEntry type="sourceFolder" forTests="false" />
7
+ </component>
8
+ </module>
.idea/inspectionProfiles/Project_Default.xml ADDED
@@ -0,0 +1,16 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ <component name="InspectionProjectProfileManager">
2
+ <profile version="1.0">
3
+ <option name="myName" value="Project Default" />
4
+ <inspection_tool class="PyPackageRequirementsInspection" enabled="true" level="WARNING" enabled_by_default="true">
5
+ <option name="ignoredPackages">
6
+ <value>
7
+ <list size="3">
8
+ <item index="0" class="java.lang.String" itemvalue="onnxruntime-gpu" />
9
+ <item index="1" class="java.lang.String" itemvalue="opencv-python" />
10
+ <item index="2" class="java.lang.String" itemvalue="imread-from-url" />
11
+ </list>
12
+ </value>
13
+ </option>
14
+ </inspection_tool>
15
+ </profile>
16
+ </component>
.idea/inspectionProfiles/profiles_settings.xml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ <component name="InspectionProjectProfileManager">
2
+ <settings>
3
+ <option name="USE_PROJECT_PROFILE" value="false" />
4
+ <version value="1.0" />
5
+ </settings>
6
+ </component>
.idea/misc.xml ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ <?xml version="1.0" encoding="UTF-8"?>
2
+ <project version="4">
3
+ <component name="ProjectRootManager" version="2" project-jdk-name="Python 3.8 (rasa2x)" project-jdk-type="Python SDK" />
4
+ </project>
.idea/modules.xml ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ <?xml version="1.0" encoding="UTF-8"?>
2
+ <project version="4">
3
+ <component name="ProjectModuleManager">
4
+ <modules>
5
+ <module fileurl="file://$PROJECT_DIR$/.idea/Uz-Text-Summarization.iml" filepath="$PROJECT_DIR$/.idea/Uz-Text-Summarization.iml" />
6
+ </modules>
7
+ </component>
8
+ </project>
.idea/vcs.xml ADDED
@@ -0,0 +1,6 @@
 
 
 
 
 
 
 
1
+ <?xml version="1.0" encoding="UTF-8"?>
2
+ <project version="4">
3
+ <component name="VcsDirectoryMappings">
4
+ <mapping directory="" vcs="Git" />
5
+ </component>
6
+ </project>
app.py ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import gradio as gr
2
+ import torch
3
+ from transformers import BertTokenizerFast, EncoderDecoderModel
4
+
5
+ device = 'cuda' if torch.cuda.is_available() else 'cpu'
6
+ ckpt = 'csebuetnlp/mT5_multilingual_XLSum'
7
+ tokenizer = BertTokenizerFast.from_pretrained(ckpt)
8
+ model = EncoderDecoderModel.from_pretrained(ckpt).to(device)
9
+
10
+ def generate_summary(text):
11
+
12
+ inputs = tokenizer([text], padding="max_length", truncation=True, max_length=512, return_tensors="pt")
13
+ input_ids = inputs.input_ids.to(device)
14
+ attention_mask = inputs.attention_mask.to(device)
15
+ output = model.generate(input_ids, attention_mask=attention_mask)
16
+ return tokenizer.decode(output[0], skip_special_tokens=True)
17
+
18
+ demo = gr.Interface(fn=generate_summary,
19
+ inputs=gr.Textbox(lines=10, placeholder="Insert the text here"),
20
+ outputs=gr.Textbox(lines=4)
21
+ )
22
+
23
+ demo.launch()
requirements.txt ADDED
@@ -0,0 +1,2 @@
 
 
 
1
+ torch
2
+ transformers