DR-Rakshitha commited on
Commit
a43c59b
1 Parent(s): afaa3f6

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -175,7 +175,7 @@ def generate_text(prompt):
175
  # prompt = input()
176
  additional_prompt = "You are an AI Medical customer care bot. Please provide detailed and complete answers for only medical questions."
177
  prompt = additional_prompt + prompt
178
- pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=1000)
179
  result = pipe(f"<s>[INST] {prompt} [/INST]")
180
  output = result[0]['generated_text']
181
  return output
 
175
  # prompt = input()
176
  additional_prompt = "You are an AI Medical customer care bot. Please provide detailed and complete answers for only medical questions."
177
  prompt = additional_prompt + prompt
178
+ pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=200)
179
  result = pipe(f"<s>[INST] {prompt} [/INST]")
180
  output = result[0]['generated_text']
181
  return output