trigger166 commited on
Commit
d4034ee
·
1 Parent(s): e5c86e6
Files changed (1) hide show
  1. app.py +12 -9
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()