CosmoAI commited on
Commit
7edbcbc
·
verified ·
1 Parent(s): d373e61

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,8 +7,10 @@ model = pipeline(model='google/flan-t5-base')
7
 
8
  # Define the chatbot function
9
  def chatbot(input_text):
 
 
10
  # Generate a response from the Hugging Face model
11
- response = model(input_text, max_length=250, do_sample=True)[0]['generated_text'].strip()
12
 
13
  # Return the bot response
14
  return response
 
7
 
8
  # Define the chatbot function
9
  def chatbot(input_text):
10
+
11
+ prompt = f"Give the answer of the given input in context from the bhagwat geeta. give suggestions to user which are based upon the meanings of shlok in bhagwat geeta, input = {input_text}"
12
  # Generate a response from the Hugging Face model
13
+ response = model(prompt, max_length=250, do_sample=True)[0]['generated_text'].strip()
14
 
15
  # Return the bot response
16
  return response