Update app.py

app.py

@@ -1,12 +1,14 @@
-import gradio as gr
+# import gradio as gr
 
 
 
-gradio_app = gr.Interface(
-    title="Advertisment companion",
-)
+# gradio_app = gr.Interface(
+#     inputs="text",
+#     outputs="text",
+#     title="Advertisment companion",
+# )
 
-gr.load("models/microsoft/Phi-3.5-mini-instruct").launch(share=True)
+# gr.load("models/microsoft/Phi-3.5-mini-instruct").launch(share=True)
 
 
 
@@ -51,3 +53,32 @@ gr.load("models/microsoft/Phi-3.5-mini-instruct").launch(share=True)
 
 # # Launch the Gradio app with sharing enabled
 # interface.launch(share=True)
+
+
+import gradio as gr
+from transformers import pipeline
+
+# Load the model pipeline for text generation
+generator = pipeline("text-generation", model="microsoft/Phi-3.5-mini-instruct", use_auth_token=True)
+
+# Define the role prompt for advertisement assistance
+role_prompt = "You are an advertisement assistant. Respond professionally and helpfully to advertising-related questions.\n\n"
+
+# Function to generate responses
+def generate_response(user_input):
+    input_text = role_prompt + user_input
+    response = generator(input_text, max_new_tokens=50, temperature=0.7, top_p=0.9)
+    return response[0]["generated_text"]
+
+# Set up Gradio interface
+interface = gr.Interface(
+    fn=generate_response,
+    inputs="text",
+    outputs="text",
+    title="Advertisement Assistant Chatbot",
+    description="Ask me anything related to advertising. I'm here to help!"
+)
+
+# Launch the Gradio app with sharing enabled
+interface.launch(share=True)
+