Sathvik-kota commited on
Commit
03cff90
·
verified ·
1 Parent(s): 95b270d

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -17,7 +17,6 @@ import cv2
17
  import google.generativeai as genai
18
 
19
  # ---------------- LLM CONFIG (Gemini) ----------------
20
-
21
  GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
22
  GEMINI_MODEL_NAME = "gemini-2.5-flash"
23
 
@@ -25,6 +24,11 @@ if GEMINI_API_KEY:
25
  genai.configure(api_key=GEMINI_API_KEY)
26
 
27
  # ---------------- FASTAPI APP ----------------
 
 
 
 
 
28
 
29
  app = FastAPI()
30
 
 
17
  import google.generativeai as genai
18
 
19
  # ---------------- LLM CONFIG (Gemini) ----------------
 
20
  GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
21
  GEMINI_MODEL_NAME = "gemini-2.5-flash"
22
 
 
24
  genai.configure(api_key=GEMINI_API_KEY)
25
 
26
  # ---------------- FASTAPI APP ----------------
27
+ app = FastAPI(title="Bajaj Datathon - Bill Extractor")
28
+
29
+ class BillRequest(BaseModel):
30
+ document: str
31
+ # ---------------- FASTAPI APP ----------------
32
 
33
  app = FastAPI()
34