sguertl commited on
Commit
8706698
·
verified ·
1 Parent(s): b0c020a

Remove test endpoint

Browse files
Files changed (1) hide show
  1. app.py +1 -5
app.py CHANGED
@@ -14,10 +14,6 @@ client = InferenceClient(model=MODEL, token=HF_TOKEN)
14
  class Prompt(BaseModel):
15
  message: str
16
 
17
- @app.get("/test")
18
- async def test():
19
- return {"Test"}
20
-
21
  @app.post("/chat")
22
  async def chat(prompt: Prompt):
23
  print("Received POST request")
@@ -38,5 +34,5 @@ async def chat(prompt: Prompt):
38
  do_sample=True
39
  )
40
 
41
- print("Text generation done")
42
  return {"reply": output.strip()}
 
14
  class Prompt(BaseModel):
15
  message: str
16
 
 
 
 
 
17
  @app.post("/chat")
18
  async def chat(prompt: Prompt):
19
  print("Received POST request")
 
34
  do_sample=True
35
  )
36
 
37
+ print("Text generation done", output.strip())
38
  return {"reply": output.strip()}