DR-Rakshitha commited on
Commit
31dda22
·
1 Parent(s): 9f3822d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -173,7 +173,7 @@ def generate_text(prompt):
173
  # return result
174
  logging.set_verbosity(logging.CRITICAL)
175
  # prompt = input()
176
- additional_prompt = "If the following question is not related to medical domain or hospital domain, Do not answer. \n"
177
  prompt = additional_prompt + prompt
178
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=100)
179
  result = pipe(f"<s>[INST] {prompt} [/INST]")
 
173
  # return result
174
  logging.set_verbosity(logging.CRITICAL)
175
  # prompt = input()
176
+ additional_prompt = "If the following question is not related to medical domain or hospital domain, Reply /'this is not my domain./'. \n"
177
  prompt = additional_prompt + prompt
178
  pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=100)
179
  result = pipe(f"<s>[INST] {prompt} [/INST]")