Prat0 commited on
Commit
cd03df2
·
verified ·
1 Parent(s): cbcfae0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -16,7 +16,7 @@ os.environ["GOOGLE_API_KEY"] = "your_api_key_here" # Replace with your actual A
16
  Settings.embed_model = GeminiEmbedding(api_key=os.getenv("GOOGLE_API_KEY"), model_name="models/embedding-001")
17
  Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.8, model_name="models/gemini-pro")
18
 
19
- def write_to_file(content, filename="test.pdf"):
20
  with open(filename, "wb") as f:
21
  f.write(content)
22
 
 
16
  Settings.embed_model = GeminiEmbedding(api_key=os.getenv("GOOGLE_API_KEY"), model_name="models/embedding-001")
17
  Settings.llm = Gemini(api_key=os.getenv("GOOGLE_API_KEY"), temperature=0.8, model_name="models/gemini-pro")
18
 
19
+ def write_to_file(content, filename="./files/test.pdf"):
20
  with open(filename, "wb") as f:
21
  f.write(content)
22