Phoenix21 commited on
Commit
3c6abe9
·
verified ·
1 Parent(s): 79ce86b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -0
app.py CHANGED
@@ -6,6 +6,13 @@ from sklearn.cluster import KMeans
6
  import networkx as nx
7
  import matplotlib.pyplot as plt
8
  import collections
 
 
 
 
 
 
 
9
 
10
  # 1. Models & Datasets Configs
11
  MODELS = ["gpt2", "distilgpt2", "qwen/Qwen2.5-0.5B", "TinyLlama/TinyLlama-1.1B-Chat-v1.0"]
 
6
  import networkx as nx
7
  import matplotlib.pyplot as plt
8
  import collections
9
+ import os
10
+ import google.generativeai as genai
11
+
12
+ # Hugging Face Secrets access
13
+ api_key = os.environ.get("GEMINI_API_KEY")
14
+ if api_key:
15
+ genai.configure(api_key=api_key)
16
 
17
  # 1. Models & Datasets Configs
18
  MODELS = ["gpt2", "distilgpt2", "qwen/Qwen2.5-0.5B", "TinyLlama/TinyLlama-1.1B-Chat-v1.0"]