Spaces:
Paused
Paused
Commit
·
a43c59b
1
Parent(s):
afaa3f6
Update app.py
Browse files
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=
|
| 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
|