shibly100 commited on
Commit
7a4012e
·
verified ·
1 Parent(s): 80193ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +10 -12
app.py CHANGED
@@ -1,28 +1,26 @@
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
- # Load your fine-tuned model from Hugging Face
5
- generator = pipeline(
6
- "text-generation",
7
- model="shibly100/gpt2_corporate_leadership"
8
- )
9
 
10
- # Define function for chatbot
11
  def chat_with_ai(prompt):
12
  response = generator(prompt, max_length=200, do_sample=True)
13
  return response[0]["generated_text"]
14
 
15
- # Create Gradio Interface
16
  demo = gr.Interface(
17
  fn=chat_with_ai,
18
- inputs="text",
19
  outputs="text",
20
  title="SPARKY Corporate Leadership AI",
21
- description="Ask anything about corporate leadership!"
 
22
  )
23
 
24
- # Launch the Gradio app
25
- if __name__ == "__main__":
26
- demo.launch()
27
 
28
 
 
1
  import gradio as gr
2
  from transformers import pipeline
3
 
4
+ # Load your fine-tuned Corporate Leadership model from Hugging Face
5
+ generator = pipeline("text-generation", model="shibly100/gpt2_corporate_leadership")
 
 
 
6
 
7
+ # Define chatbot function
8
  def chat_with_ai(prompt):
9
  response = generator(prompt, max_length=200, do_sample=True)
10
  return response[0]["generated_text"]
11
 
12
+ # Create the Gradio Chatbot Interface
13
  demo = gr.Interface(
14
  fn=chat_with_ai,
15
+ inputs=gr.Textbox(lines=3, placeholder="Ask anything about corporate leadership..."),
16
  outputs="text",
17
  title="SPARKY Corporate Leadership AI",
18
+ description="Ask anything about corporate leadership and get AI-powered insights!",
19
+ allow_flagging="never",
20
  )
21
 
22
+ # Launch the chatbot
23
+ demo.launch()
24
+
25
 
26