matteomarjanovic commited on
Commit
42b1e2e
·
1 Parent(s): e94fb4a

first commit google gemini

Browse files
Files changed (2) hide show
  1. app.py +3 -0
  2. 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