login

app.py

@@ -1,13 +1,23 @@
+import os
 import gradio as gr
 import torch
-from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, Trainer, TrainingArguments
+from transformers import AutoModelForSeq2SeqLM, AutoTokenizer, Trainer, TrainingArguments, EncoderDecoderCache
 from datasets import load_dataset
+from huggingface_hub import login
+
+
+hf_token = os.getenv("HF_TOKEN")
+if hf_token is None:
+    raise ValueError("Il token HF_TOKEN non è impostato nelle variabili d'ambiente")
+login(hf_token)
+
+
 
 # Carica il dataset dal file JSONL
 dataset = load_dataset("json", data_files="data.jsonl")
 
 # Definisci il modello e il tokenizer
-model_name = "google/flan-t5-small"  # Puoi scegliere un modello T5 base per semplicità
+model_name = "google/flan-t5-small" # Puoi scegliere un modello T5 base per semplicità
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModelForSeq2SeqLM.from_pretrained(model_name)