WICKED4950 commited on
Commit
d6a99aa
·
verified ·
1 Parent(s): 07f8c16

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
- # Load your model
 
 
 
4
  print("Loading the model......")
5
  model_name = "WICKED4950/Irisonego5"
6
  tokenizer = AutoTokenizer.from_pretrained(model_name)
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
+ from huggingface_hub import login, create_repo, upload_file
4
+
5
+ # Log in to Hugging Face
6
+ login(hf_token)
7
  print("Loading the model......")
8
  model_name = "WICKED4950/Irisonego5"
9
  tokenizer = AutoTokenizer.from_pretrained(model_name)