Spaces:

Andolinism
/

TelegramChatSummarizer

Runtime error

TelegramChatSummarizer / summarizer /models.py

Kaveh

Update models.py

1bd2562 unverified over 1 year ago

403 Bytes

from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

def load_model():
    model_name = "nafisehNik/mt5-persian-summary"
    cache_dir = "/tmp/hf_cache"  # مسیر مجاز برای Hugging Face Spaces
    tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=cache_dir)
    model = AutoModelForSeq2SeqLM.from_pretrained(model_name, cache_dir=cache_dir)
    return tokenizer, model

	from transformers import AutoTokenizer, AutoModelForSeq2SeqLM

	def load_model():
	model_name = "nafisehNik/mt5-persian-summary"
	cache_dir = "/tmp/hf_cache" # مسیر مجاز برای Hugging Face Spaces
	tokenizer = AutoTokenizer.from_pretrained(model_name, cache_dir=cache_dir)
	model = AutoModelForSeq2SeqLM.from_pretrained(model_name, cache_dir=cache_dir)
	return tokenizer, model