Hugging Face's logo

Spaces:
K00B404
/
Mistral-Nemo-custom
Running

App Files Community
K00B404 commited on
Commit
02b7231
·
verified ·
1 Parent(s): 64b62db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -31,7 +31,7 @@ h3 {
31
  }
32
  """
33
 
34
- device = "cuda" # for GPU usage or "cpu" for CPU usage
35
 
36
  tokenizer = AutoTokenizer.from_pretrained(MODEL)
37
  model = AutoModelForCausalLM.from_pretrained(
 
31
  }
32
  """
33
 
34
+ device = "cpu" # for GPU usage or "cpu" for CPU usage
35
 
36
  tokenizer = AutoTokenizer.from_pretrained(MODEL)
37
  model = AutoModelForCausalLM.from_pretrained(