jjz5463 commited on
Commit
80db327
1 Parent(s): c21d500
Files changed (2) hide show
  1. app.py +6 -0
  2. baseline_utils.py +1 -1
app.py CHANGED
@@ -12,6 +12,12 @@ openai_api_key = os.getenv("OPENAI_API_KEY")
12
  google_service_account_info = json.loads(os.getenv("GOOGLE_SERVICE_ACCOUNT"))
13
  gemini_api_key = os.getenv("GEMINI_API_KEY")
14
 
 
 
 
 
 
 
15
 
16
  # Function to get Google credentials
17
  def get_google_credentials():
 
12
  google_service_account_info = json.loads(os.getenv("GOOGLE_SERVICE_ACCOUNT"))
13
  gemini_api_key = os.getenv("GEMINI_API_KEY")
14
 
15
+ # from keys.keys import *
16
+ # # Load secrets from the environment or other sources (adjust as needed)
17
+ # openai_api_key = open_ai_keys
18
+ # with open('keys/service_account_credentials.json') as f:
19
+ # google_service_account_info = json.load(f)
20
+ # gemini_api_key = gemini_keys
21
 
22
  # Function to get Google credentials
23
  def get_google_credentials():
baseline_utils.py CHANGED
@@ -79,7 +79,7 @@ def generate_comic_book(diary_text, writer_description, num_pages=4):
79
  "stabilityai/sdxl-turbo",
80
  torch_dtype=torch.float16,
81
  variant="fp16",
82
- #cache_dir="./SDXL-Turbo"
83
  )
84
 
85
  # Check for available device: CUDA, MPS, or CPU
 
79
  "stabilityai/sdxl-turbo",
80
  torch_dtype=torch.float16,
81
  variant="fp16",
82
+ cache_dir="./SDXL-Turbo"
83
  )
84
 
85
  # Check for available device: CUDA, MPS, or CPU