Upload app.py

app.py

@@ -1,29 +1,19 @@
-import gradio as gr
-from transformers import pipeline
-
-# Load the fine-tuned Corporate Leadership model from Hugging Face
-generator = pipeline("text-generation", model="shibly100/gpt2_corporate_leadership")
-
-# Define the chat function that remembers conversation history
-def chat_with_ai(history, message):
-    full_prompt = " ".join([turn[0] + " " + turn[1] for turn in history] + [message])
-    response = generator(full_prompt, max_length=300, do_sample=True, temperature=0.7)
-    reply = response[0]["generated_text"]
-    
-    # Ensure only new AI-generated response is returned
-    reply = reply[len(full_prompt):].strip()
-    history.append((message, reply))
-    return history, ""
-
-# Create Chatbot Interface
-chatbot = gr.ChatInterface(
-    chat_with_ai,
-    title="🗣️ SPARKY Corporate Leadership AI",
-    description="🤖 Ask anything about corporate leadership and get AI-powered responses! Your conversation history is remembered.",
-    theme="soft",  # This gives it a sleek, modern look like ChatGPT
-)
-
-# Launch the chatbot
-chatbot.launch()
-
- 
+import gradio as gr
+from transformers import pipeline
+
+# Load the fine-tuned model
+generator = pipeline("text-generation", model="shibly100/gpt2_corporate_leadership")
+
+# Function to generate responses
+def chat_with_ai(prompt, history=[]):
+    response = generator(prompt, max_length=200, do_sample=True, top_p=0.95, top_k=50, temperature=0.7)
+    return response[0]["generated_text"]
+
+# Fast and simple chat interface
+with gr.Blocks() as demo:
+    gr.Markdown("# SPARKY Corporate Leadership AI")
+    gr.Markdown("Ask anything about corporate leadership and get AI-powered responses!")
+
+    chatbot = gr.ChatInterface(chat_with_ai)
+
+demo.launch()