Spaces:
Sleeping
Sleeping
Commit
·
42b1e2e
1
Parent(s):
e94fb4a
first commit google gemini
Browse files- app.py +3 -0
- requirements.txt +2 -1
app.py
CHANGED
@@ -14,6 +14,9 @@ import base64
|
|
14 |
from io import BytesIO
|
15 |
import os
|
16 |
|
|
|
|
|
|
|
17 |
login(token=os.environ.get("HF_API_KEY"))
|
18 |
|
19 |
subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
|
|
|
14 |
from io import BytesIO
|
15 |
import os
|
16 |
|
17 |
+
from google import genai
|
18 |
+
from google.genai import types
|
19 |
+
|
20 |
login(token=os.environ.get("HF_API_KEY"))
|
21 |
|
22 |
subprocess.run("rm -rf /data-nvme/zerogpu-offload/*", env={}, shell=True)
|
requirements.txt
CHANGED
@@ -8,4 +8,5 @@ sentencepiece
|
|
8 |
peft
|
9 |
groq
|
10 |
controlnet-aux
|
11 |
-
mediapipe
|
|
|
|
8 |
peft
|
9 |
groq
|
10 |
controlnet-aux
|
11 |
+
mediapipe
|
12 |
+
google-genai
|