Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -8,6 +8,7 @@ import gradio as gr
|
|
8 |
|
9 |
text_generator = None
|
10 |
is_hugging_face = True
|
|
|
11 |
def init():
|
12 |
global text_generator
|
13 |
huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
@@ -16,7 +17,7 @@ def init():
|
|
16 |
print("no HUGGINGFACE_TOKEN if you need set secret ")
|
17 |
#raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
|
18 |
|
19 |
-
|
20 |
#model_id = "microsoft/Phi-3-mini-128k-instruct"
|
21 |
|
22 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
|
|
8 |
|
9 |
text_generator = None
|
10 |
is_hugging_face = True
|
11 |
+
model_id = "google/gemma-2-9b-it"
|
12 |
def init():
|
13 |
global text_generator
|
14 |
huggingface_token = os.getenv("HUGGINGFACE_TOKEN")
|
|
|
17 |
print("no HUGGINGFACE_TOKEN if you need set secret ")
|
18 |
#raise ValueError("HUGGINGFACE_TOKEN environment variable is not set")
|
19 |
|
20 |
+
|
21 |
#model_id = "microsoft/Phi-3-mini-128k-instruct"
|
22 |
|
23 |
device = "auto" # torch.device("cuda" if torch.cuda.is_available() else "cpu")
|