gregH commited on
Commit
dece9b2
1 Parent(s): 197924f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -43,6 +43,8 @@ print(f"Starting to load the model to memory")
43
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
44
 
45
  HF_TOKEN = os.getenv("HF_TOKEN")
 
 
46
 
47
  m = AutoModelForCausalLM.from_pretrained(
48
  "google/gemma-2b-it",
 
43
  device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
44
 
45
  HF_TOKEN = os.getenv("HF_TOKEN")
46
+ prnit("HF_TOKEN")
47
+
48
 
49
  m = AutoModelForCausalLM.from_pretrained(
50
  "google/gemma-2b-it",