Spaces:
Sleeping
Sleeping
Remove test endpoint
Browse files
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()}
|