PRITHVI-V commited on
Commit
a97ab17
1 Parent(s): 3bb8124

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -13,6 +13,7 @@ genai.configure(api_key=os.environ["API_KEY"])
13
  hf_token = os.getenv("HF_TOK") # Set your Hugging Face token in your environment variables
14
  login(hf_token) # Log in using the Hugging Face token
15
 
 
16
  model = genai.GenerativeModel("gemini-1.5-flash")
17
 
18
  # Load the pre-trained model for Tamil to English translation
 
13
  hf_token = os.getenv("HF_TOK") # Set your Hugging Face token in your environment variables
14
  login(hf_token) # Log in using the Hugging Face token
15
 
16
+ diffusion_model = InferenceApi(repo_id="black-forest-labs/FLUX.1-schnell")
17
  model = genai.GenerativeModel("gemini-1.5-flash")
18
 
19
  # Load the pre-trained model for Tamil to English translation