init

app.py

@@ -39,7 +39,7 @@ def fine_tune_model(model_name, dataset_name, hub_id, api_key, num_epochs, batch
     
         # Tokenize the dataset
         def tokenize_function(examples):
-            max_length = 32
+            max_length = 16
             # Assuming 'text' is the input and 'target' is the expected output
             model_inputs = tokenizer(examples['text'], max_length=max_length, truncation=True)
         
@@ -110,7 +110,7 @@ try:
             gr.Textbox(label="HF API token"),
             gr.Slider(minimum=1, maximum=10, value=3, label="Number of Epochs", step=1),
             gr.Slider(minimum=1, maximum=16, value=1, label="Batch Size", step=1),
-            gr.Slider(minimum=1, maximum=1000, value=1, label="Learning Rate (e-5)", step=1),
+            gr.Slider(minimum=1, maximum=1000, value=1, label="Learning Rate (e-6)", step=1),
             gr.Slider(minimum=1, maximum=100, value=1, label="Gradient accumulation (e-1)", step=1), 
         ],
         outputs="text",