Pulastya0 commited on
Commit
025579b
·
verified ·
1 Parent(s): f8a0e18

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,6 +1,6 @@
1
  import os
2
  from fastapi import FastAPI, HTTPException
3
- from pydantic import BaseModel
4
  from transformers import pipeline
5
 
6
  # -------------------------------
@@ -21,6 +21,9 @@ app = FastAPI(title="Routing Service - Space 2")
21
  class RoutingRequest(BaseModel):
22
  text: str
23
 
 
 
 
24
  # -------------------------------
25
  # Load Zero-Shot Classifier
26
  # -------------------------------
 
1
  import os
2
  from fastapi import FastAPI, HTTPException
3
+ from pydantic import BaseModel , Extra
4
  from transformers import pipeline
5
 
6
  # -------------------------------
 
21
  class RoutingRequest(BaseModel):
22
  text: str
23
 
24
+ class Config:
25
+ extra = Extra.ignore # allow extra fields like user_email
26
+
27
  # -------------------------------
28
  # Load Zero-Shot Classifier
29
  # -------------------------------