John6666 commited on
Commit
7d1fab1
·
verified ·
1 Parent(s): 9a2f119

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from transformers import pipeline
7
  #model_name = "RickyDeSkywalker/TheoremLlama"
8
  model_name = "unsloth/Llama-3.2-1B-Instruct"
9
  HF_TOKEN = os.environ.get("HF_TOKEN")
10
- login(HF_TOKEN)
11
 
12
  generator = pipeline('text-generation', model=model_name, token=HF_TOKEN)
13
 
 
7
  #model_name = "RickyDeSkywalker/TheoremLlama"
8
  model_name = "unsloth/Llama-3.2-1B-Instruct"
9
  HF_TOKEN = os.environ.get("HF_TOKEN")
10
+ #login(HF_TOKEN)
11
 
12
  generator = pipeline('text-generation', model=model_name, token=HF_TOKEN)
13