Spaces:
Runtime error
Runtime error
Commit ·
d4034ee
1
Parent(s): e5c86e6
no eval
Browse files
app.py
CHANGED
|
@@ -26,6 +26,7 @@ tokenized_dataset = dataset.map(preprocess_data, batched=True)
|
|
| 26 |
training_args = TrainingArguments(
|
| 27 |
output_dir="./results",
|
| 28 |
evaluation_strategy="epoch",
|
|
|
|
| 29 |
learning_rate=2e-5,
|
| 30 |
per_device_train_batch_size=4,
|
| 31 |
num_train_epochs=3,
|
|
@@ -39,13 +40,15 @@ trainer = Trainer(
|
|
| 39 |
train_dataset=tokenized_dataset["train"]
|
| 40 |
)
|
| 41 |
|
|
|
|
|
|
|
| 42 |
# Funzione di Gradio per effettuare previsioni
|
| 43 |
-
def answer_question(question):
|
| 44 |
-
inputs = tokenizer(question, return_tensors="pt", padding="max_length", truncation=True, max_length=128)
|
| 45 |
-
outputs = model.generate(inputs["input_ids"], max_length=50)
|
| 46 |
-
answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 47 |
-
return answer
|
| 48 |
-
|
| 49 |
-
# Interfaccia Gradio
|
| 50 |
-
iface = gr.Interface(fn=answer_question, inputs="text", outputs="text")
|
| 51 |
-
iface.launch()
|
|
|
|
| 26 |
training_args = TrainingArguments(
|
| 27 |
output_dir="./results",
|
| 28 |
evaluation_strategy="epoch",
|
| 29 |
+
eval_strategy="no", # Disabilita la valutazione durante l'addestramento
|
| 30 |
learning_rate=2e-5,
|
| 31 |
per_device_train_batch_size=4,
|
| 32 |
num_train_epochs=3,
|
|
|
|
| 40 |
train_dataset=tokenized_dataset["train"]
|
| 41 |
)
|
| 42 |
|
| 43 |
+
trainer.train()
|
| 44 |
+
|
| 45 |
# Funzione di Gradio per effettuare previsioni
|
| 46 |
+
#def answer_question(question):
|
| 47 |
+
# inputs = tokenizer(question, return_tensors="pt", padding="max_length", truncation=True, max_length=128)
|
| 48 |
+
# outputs = model.generate(inputs["input_ids"], max_length=50)
|
| 49 |
+
# answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
|
| 50 |
+
# return answer
|
| 51 |
+
#
|
| 52 |
+
## Interfaccia Gradio
|
| 53 |
+
#iface = gr.Interface(fn=answer_question, inputs="text", outputs="text")
|
| 54 |
+
#iface.launch()
|