tanya17 commited on
Commit
f0af713
·
verified ·
1 Parent(s): a72a860

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -15,8 +15,10 @@ import random
15
  import requests
16
  import os
17
 
18
- # Hugging Face API configuration
19
- HUGGINGFACE_API_URL = "https://api-inference.huggingface.co/models/google/flan-t5-large"
 
 
20
  HUGGINGFACE_API_KEY = os.environ.get("HUGGINGFACE_API_KEY")
21
 
22
  # File to store feedback
 
15
  import requests
16
  import os
17
 
18
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
19
+
20
+ tokenizer = AutoTokenizer.from_pretrained("google/flan-t5-base")
21
+ model = AutoModelForSeq2SeqLM.from_pretrained("google/flan-t5-base")
22
  HUGGINGFACE_API_KEY = os.environ.get("HUGGINGFACE_API_KEY")
23
 
24
  # File to store feedback