Joetib commited on
Commit
9554a41
1 Parent(s): be6e00e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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>What programmes are offered at fordham university?</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']}