Spaces:
Runtime error
Runtime error
Commit
•
5be871e
1
Parent(s):
f440eb0
Update main.py
Browse files
main.py
CHANGED
|
@@ -20,7 +20,7 @@ from transformers import AutoTokenizer, TextStreamer, pipeline
|
|
| 20 |
|
| 21 |
DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 22 |
|
| 23 |
-
os.environ['GOOGLE_API_KEY']
|
| 24 |
|
| 25 |
genai.configure(api_key=GOOGLE_API_KEY)
|
| 26 |
model = genai.GenerativeModel('gemini-pro')
|
|
|
|
| 20 |
|
| 21 |
DEVICE = "cuda:0" if torch.cuda.is_available() else "cpu"
|
| 22 |
|
| 23 |
+
GOOGLE_API_KEY = os.environ['GOOGLE_API_KEY']
|
| 24 |
|
| 25 |
genai.configure(api_key=GOOGLE_API_KEY)
|
| 26 |
model = genai.GenerativeModel('gemini-pro')
|