Update app.py

app.py

@@ -107,11 +107,11 @@ trainer = SFTTrainer(
     train_dataset=dataset,
     dataset_text_field="text",
     max_seq_length=max_seq_length,
-    dataset_num_proc=2,
+    dataset_num_proc=20,
     packing=False,
     args=TrainingArguments(
-        per_device_train_batch_size=2,
-        gradient_accumulation_steps=2,
+        per_device_train_batch_size=20,
+        gradient_accumulation_steps=20,
         learning_rate=2e-4,
         fp16=not is_bfloat16_supported(),
         bf16=is_bfloat16_supported(),
@@ -141,13 +141,4 @@ model.push_to_hub_merged(
     save_method="merged_16bit",
     token=True
 )
-print("Model pushed to hub successfully.")
-
-# Gradio app
-print("Launching Gradio app...")
-def greet(name):
-    return "Hello " + name + "!!"
-
-demo = gr.Interface(fn=greet, inputs="text", outputs="text")
-demo.launch()
-print("Gradio app launched.")
+print("Model pushed to hub successfully.")