Update app.py
Browse files
app.py
CHANGED
|
@@ -25,6 +25,6 @@ async def root():
|
|
| 25 |
@app.get("/generate/")
|
| 26 |
async def generate_text(prompt: str):
|
| 27 |
# Generate text based on the prompt received
|
| 28 |
-
prompt = "<s>
|
| 29 |
response = pipeline(prompt)
|
| 30 |
return {"generated_text": response['generated_text']}
|
|
|
|
| 25 |
@app.get("/generate/")
|
| 26 |
async def generate_text(prompt: str):
|
| 27 |
# Generate text based on the prompt received
|
| 28 |
+
prompt = f"<s>[INST]{prompt}</INST>"
|
| 29 |
response = pipeline(prompt)
|
| 30 |
return {"generated_text": response['generated_text']}
|