Update app.py

app.py

@@ -173,9 +173,9 @@ 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 reply."
+     additional_prompt = "You are an AI Medical customer care bot. Please provide detailed and complete answers for only medical questions."
      prompt = additional_prompt + prompt
-     pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer)
+     pipe = pipeline(task="text-generation", model=model, tokenizer=tokenizer, max_length=1000)
      result = pipe(f"<s>[INST] {prompt} [/INST]")
      output = result[0]['generated_text']
      return output