Spaces:

injilashah
/

Bloom_vs_gemma

Sleeping

injilashah commited on Dec 11, 2024

Commit

6d9aa31

verified ·

1 Parent(s): 0a05bf7

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -2,7 +2,7 @@ import os
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
-hf_token = os.getenv("hf_token2")
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")

 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import gradio as gr
+hf_token = os.environ["hf_token"]
 device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")