Update Home.py

Home.py

@@ -2,7 +2,8 @@ import streamlit as st
 from streamlit_option_menu import option_menu
 from transformers import pipeline, Conversation
 
-pipe = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
+convo = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
+imgclassifier = pipeline(model="microsoft/beit-base-patch16-224-pt22k-ft22k")
 
 
 # def homepage():
@@ -20,14 +21,18 @@ pipe = pipeline(task="conversational", model="microsoft/DialoGPT-medium")
 def chat():
     if query := st.chat_input("Enter your message"):
         uquery = Conversation(query)
-        response = pipe(uquery)
+        response = convo(uquery)
         with st.chat_message("assistant"):
             st.write(response.generated_responses[-1])
         
     
-# def invoke_document():
-#     st.write("Invoke Document")
-#     st.write("Welcome to the invoke document page")
+def image_classifi():
+    with st.sidebar:
+        file = st.file_uploader("Upload Image", ["png", "jpg"])
+    output = imgclassifier(file)
+    if st.button("View Results"):
+        st.write(output)
+        
     
 # def invoke_audio():
 #     st.write("Invoke Audio")