networkbot / config.py
Adieva-15's picture
fixed generate.py
470297a
Raw
History Blame Contribute Delete
571 Bytes
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
import os
os.environ['HF_ENDPOINT']='https://hf-mirror.com'
BOT_TOKEN = os.getenv("BOT_TOKEN")
HF_TOKEN = os.getenv("HF_TOKEN")
SENTIMENT_API = "https://huggingface.co/tabularisai/multilingual-sentiment-analysis"
model_name = "tabularisai/multilingual-sentiment-analysis"
SUMMARIZE_API=''
model_sum='IlyaGusev/rut5_base_sum_gazeta'
EMOTION_API = "https://api-inference.huggingface.co/models/harsh3474/face-emotion-recognition"
headers = {"Authorization": f"Bearer {HF_TOKEN}"}