Hugging Face's logo

Spaces:
SimpleBerry
/
LLaMA-O1-Supervised-1129-Demo
Running

App Files Community
1
Di Zhang commited on
Commit
0aa2267
·
verified ·
1 Parent(s): 57d9030

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import spaces
4
  import os
5
  import gradio as gr
6
  from transformers import LlamaForCausalLM, LlamaTokenizer
7
- from huggingface_hub import hf_hub_download
8
 
9
  # Load the model and tokenizer from Hugging Face
10
  model_path = snapshot_download(
 
4
  import os
5
  import gradio as gr
6
  from transformers import LlamaForCausalLM, LlamaTokenizer
7
+ from huggingface_hub import hf_hub_download, snapshot_download
8
 
9
  # Load the model and tokenizer from Hugging Face
10
  model_path = snapshot_download(