prasanth345 commited on
Commit
e1b58a1
·
verified ·
1 Parent(s): 759dbe3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -4,14 +4,14 @@ import os
4
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
5
 
6
  # API keys for other features (optional)
7
- Image_Token = os.getenv('Image_generation')
8
  Content_Token = os.getenv('Content_generation','').strip()
9
- Image_prompt_token = os.getenv('Prompt_generation')
10
 
11
  # API Headers for external services (optional)
12
  Image_generation = {"Authorization": f"Bearer {Image_Token}"}
13
  Content_generation = {
14
- "Authorization": f"Bearer {Content_Token.strip()}",
15
  "Content-Type": "application/json"
16
  }
17
  Image_Prompt = {
 
4
  from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
5
 
6
  # API keys for other features (optional)
7
+ Image_Token = os.getenv('Image_generation').strip()
8
  Content_Token = os.getenv('Content_generation','').strip()
9
+ Image_prompt_token = os.getenv('Prompt_generation').strip()
10
 
11
  # API Headers for external services (optional)
12
  Image_generation = {"Authorization": f"Bearer {Image_Token}"}
13
  Content_generation = {
14
+ "Authorization": f"Bearer {Content_Token}",
15
  "Content-Type": "application/json"
16
  }
17
  Image_Prompt = {