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']}
|