daniloedu commited on
Commit
3cca5d2
·
1 Parent(s): 38124bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -2,9 +2,13 @@ import os
2
  import requests
3
  import gradio as gr
4
  from dotenv import load_dotenv
 
5
 
6
  load_dotenv()
7
 
 
 
 
8
  API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
9
  headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
10
 
 
2
  import requests
3
  import gradio as gr
4
  from dotenv import load_dotenv
5
+ from transformers import AutoTokenizer
6
 
7
  load_dotenv()
8
 
9
+ model_name = "tiiuae/falcon-7b-instruct"
10
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
11
+
12
  API_URL = "https://api-inference.huggingface.co/models/tiiuae/falcon-7b-instruct"
13
  headers = {"Authorization": f"Bearer {os.getenv('HF_API_KEY')}"}
14