obx0x3 commited on
Commit
51c50dc
·
verified ·
1 Parent(s): ccc63ca

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -10,7 +10,7 @@ model = AutoModelWithLMHead.from_pretrained("mrm8488/t5-base-finetuned-emotion")
10
 
11
  class PromptRequest(BaseModel):
12
  message: str
13
-
14
  @app.post("/generate")
15
  async def generate_response(payload: PromptRequest):
16
  input_text = f"emotion: {payload.message}"
 
10
 
11
  class PromptRequest(BaseModel):
12
  message: str
13
+
14
  @app.post("/generate")
15
  async def generate_response(payload: PromptRequest):
16
  input_text = f"emotion: {payload.message}"