functionalities

Home.py

@@ -4,6 +4,11 @@ from transformers import pipeline, Conversation
 
 convo = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
 imgclassifier = pipeline(model="microsoft/beit-base-patch16-224-pt22k-ft22k")
+qnabot = pipeline(task="question-answering", model="distilbert-base-cased-distilled-squad")
+txtgen = pipeline(task="text-generation", model="EleutherAI/gpt-neo-2.7B")
+txtclassifi = pipeline(task="text-classification", model="nlptown/bert-base-multilingual-uncased-sentiment")
+summurize = pipeline(task="summarization", model="sshleifer/distilbart-cnn-12-6")
+visualqna = pipeline(task="vqa", model="microsoft/DialoGPT-medium")
 
 
 # def homepage():
@@ -19,6 +24,7 @@ imgclassifier = pipeline(model="microsoft/beit-base-patch16-224-pt22k-ft22k")
     
     
 def chat():
+    st.title("Chit-Chatbot")
     if query := st.chat_input("Enter your message"):
         uquery = Conversation(query)
         response = convo(uquery)
@@ -27,29 +33,52 @@ def chat():
         
     
 def image_classifi():
-    with st.sidebar:
-        file = st.file_uploader("Upload Image", ["png", "jpg"])
+    st.title("Image Classification")
+    file = st.text_input("Enter Image URL")
     output = imgclassifier(file)
     if st.button("View Results"):
         st.write(output)
         
     
-# def invoke_audio():
-#     st.write("Invoke Audio")
-#     st.write("Welcome to the invoke audio page")
-    
-# def invoke_video():
-#     st.write("Invoke Video")
-#     st.write("Welcome to the invoke video page")
+def qna_bot():
+    st.title("Q&A-Chatbot")
+    if query := st.chat_input("Enter your message"):
+        response = qnabot(query)
+        with st.chat_message("assistant"):
+            st.write(response)
+        
     
-# def invoke_image():
-#     st.write("Invoke Image")
-#     st.write("Welcome to the invoke image page")
+def txt_gen():
+    st.title("Text Generation")
+    if query := st.chat_input("Enter your message"):
+        response = txtgen(query)
+        with st.chat_message("assistant"):
+            st.write(response)
     
-# def invoke_text():
-#     st.write("Invoke Text")
-#     st.write("Welcome to the invoke text page")
+def txt_classifi():
+    st.title("Text Classification")
+    if query := st.chat_input("Enter your message"):
+        response = txtclassifi(query,)
+        with st.chat_message("assistant"):
+            st.write(response)
     
+def summury():
+    st.title("Summury")
+    if query := st.chat_input("Enter your message"):
+        response = summurize(query, min_length=5, max_length=20)
+        with st.chat_message("assistant"):
+            st.write(response)
+
+def visual_qna():
+    st.title("Visual Q&A")
+    with st.sidebar:
+        if img := st.text_input("Enter Image URL"):
+            st.image(img)
+    if query := st.chat_input("Enter your message"):
+        response = visualqna(img, query)
+    with st.chat_message("assistant"):
+        st.write(response)
+