Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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"]
|