Update app.py

app.py

@@ -173,6 +173,8 @@ def generate_text(prompt):
 #    return result
      logging.set_verbosity(logging.CRITICAL)
 #    prompt = input()
+     additional_prompt = "If the following question is not related to medical domain or hospital domain, Do not answer. \n"
+     prompt = additional_prompt + prompt
      pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=100)
      result = pipe(f"<s>[INST] {prompt} [/INST]")
      output = result[0]['generated_text']