Update app.py

app.py

@@ -1,24 +1,17 @@
 import streamlit as st
-from streamlit_tags import st_tags, st_tags_sidebar
-from keytotext import pipeline
-from PIL import Image
+from transformers import pipeline
+from keybert import KeyBERT
+from transformers.pipelines import pipeline
 
-import json
-import gzip
-import os
-import torch
-import pickle
-import yake
+st.title('Title Summarization')
 
-############
-## Main page
-############
 
-st.write("# Demonstration for Etsy Title Summarization（Etsy-TS）")
+path='checkpoint-210000'
+hf_model = pipeline("feature-extraction",model=path,tokenizer='bert-base-uncased' )
+etsy_model = KeyBERT(model=hf_model)l=hf_model)
 
-st.markdown("***Idea is to build a model which will automatically summarize title information.***")
-image = Image.open('temp.jpg')
-st.image(image)
-
-user_query = st.text_input("Enter a title for the summarization:")
+text = st.text_area("Enter title here for the summarization:")
 
+if text:
+    out = pipe(text)
+    st.json(out)