CosmoAI commited on
Commit
8c76e53
1 Parent(s): a1e4869

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -0
app.py CHANGED
@@ -1,6 +1,9 @@
1
  import gradio
2
  # from transformers import pipeline
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
 
 
 
4
 
5
  # Initialize the Hugging Face model
6
  # model = pipeline(model='google/flan-t5-base')
 
1
  import gradio
2
  # from transformers import pipeline
3
  from transformers import AutoTokenizer, AutoModelForCausalLM
4
+ import os
5
+
6
+ os.getenv("HF_TOKEN")
7
 
8
  # Initialize the Hugging Face model
9
  # model = pipeline(model='google/flan-t5-base')