Update main.py
Browse files
main.py
CHANGED
@@ -30,7 +30,7 @@ class InputData(BaseModel):
|
|
30 |
json_prompt: str
|
31 |
history: str = ""
|
32 |
|
33 |
-
|
34 |
async def generate_response(data: InputData) -> Any:
|
35 |
client = InferenceClient(model=data.model, token=HF_TOKEN)
|
36 |
|
|
|
30 |
json_prompt: str
|
31 |
history: str = ""
|
32 |
|
33 |
+
@app.post("/generate-response/")
|
34 |
async def generate_response(data: InputData) -> Any:
|
35 |
client = InferenceClient(model=data.model, token=HF_TOKEN)
|
36 |
|