entertainment title app done

app.py

@@ -1,9 +1,31 @@
+from fastai.text.all import *
+from blurr.text.modeling.all import *
 import gradio as gr
 
 
-def greet(name):
-    return "Hello " + name + "!!"
+if platform.system() == "Windows":
+    import pathlib
+    temp = pathlib.PosixPath
+    pathlib.PosixPath = pathlib.WindowsPath
 
 
-iface = gr.Interface(fn=greet, inputs="text", outputs="text")
-iface.launch()
+model_path = "entertainment-title-stage-1.pkl"
+model = load_learner(model_path)
+
+
+def entertainment_title(description):
+    outputs = model.blurr_summarize(description, early_stopping=True, num_beams=20, num_return_sequences=5)
+    return {key: 1 for key in list(outputs[0].values())[0]}
+
+
+example = [
+    ["When the menace known as the Joker wreaks havoc and chaos on the people of Gotham, Batman must accept one of "
+     "the greatest psychological and physical tests of his ability to fight injustice."],
+    ["The story of American scientist, J. Robert Oppenheimer, and his role in the development of the atomic."]
+]
+
+
+labels = gr.outputs.Label(num_top_classes=5)
+iface = gr.Interface(fn=entertainment_title, inputs="text", outputs=labels, examples=example)
+iface.interpretation = "default"
+iface.launch(inline=False, share=True)

entertainment-title-stage-1.pkl

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6278ad7a89c9571fe8030c296f9fd385ad62a496a9fb56c1ab7ac822e106e234
+size 246483451

requirements.txt

@@ -1 +1,5 @@
-gradio
+fastai==2.7.12
+gradio==3.44.2
+ohmeow-blurr==1.0.5
+sentencepiece==0.1.99
+transformers==4.33.1