Akjava commited on
Commit
ffe736b
Β·
verified Β·
1 Parent(s): 4be0fb8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
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
- model_id = "google/gemma-2-9b-it"
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")