api

app.py

@@ -46,14 +46,15 @@ training_args = TrainingArguments(
 )
 
 # Inizializza il Trainer
-trainer = Trainer(
-    model=model,
-    args=training_args,
-    train_dataset=tokenized_dataset["train"]
-)
-
-trainer.train()
-trainer.push_to_hub("testA")
+def start_training():
+    trainer = Trainer(
+        model=model,
+        args=training_args,
+        train_dataset=tokenized_dataset["train"]
+    )
+    trainer.train()
+    trainer.push_to_hub("to_validate_model")
+    return "Training completato e caricato"
 
 def answer_question(question):
     inputs = tokenizer(question, return_tensors="pt", truncation=True, max_length=128).to(device)
@@ -62,4 +63,8 @@ def answer_question(question):
     return answer
 
 iface = gr.Interface(fn=answer_question, inputs="text", outputs="text")
+train_interface = gr.Interface(fn=start_training, inputs=[], outputs="text")
+app = gr.TabbedInterface([iface, train_interface], ["Q&A", "Avvia Training"])
+
+
 iface.launch()