minor fixes to app.py

app.py

@@ -48,12 +48,13 @@ class PrinterCallback(TrainerCallback):
     def on_log(self, args, state, control, logs=None, **kwargs):
         _ = logs.pop("total_flos", None)
         if state.is_local_process_zero:
-            print(logs)
+            #print(logs)
+            pass
     def on_step_end(self, args, state, control, **kwargs):
         if state.is_local_process_zero:
             self.step = state.global_step
             self.progress.update(self.step)
-            print("**Step ", state.global_step)
+            #print("**Step ", state.global_step)
 
 
 
@@ -68,8 +69,6 @@ def formatting_prompts_func(examples, prompt):
         text = prompt.format(instruction, input, output) + EOS_TOKEN
         texts.append(text)
     return { "text" : texts, }
-pass
-
 
 def load_model(initial_model_name, load_in_4bit, max_sequence_length):
     global model, tokenizer, max_seq_length
@@ -187,11 +186,11 @@ def save_model(model_name, hub_model_name, hub_token, gguf_16bit, gguf_8bit, ggu
     return "Model saved", gr.update(visible=True, interactive=True)
 
 # Create the Gradio interface
-with gr.Blocks() as demo:
+with gr.Blocks(title="Unsloth fine-tuning") as demo:
     with gr.Column():
         gr.Image("unsloth.png", width="300px", interactive=False, show_download_button=False, show_label=False)
     with gr.Column():
-        gr.Markdown(f"**GPU Information:** {gpu_stats.name} ({max_memory} GB)") 
+        gr.Markdown(f"**GPU Information:** {gpu_stats.name} ({max_memory} GB)\n\n[Unsloth Docs](http://docs.unsloth.com/)\n\n[Unsloth GitHub](https://github.com/unslothai/unsloth)") 
     with gr.Tab("Base Model Parameters"):
 
         with gr.Row():
@@ -299,17 +298,17 @@ with gr.Blocks() as demo:
                 hub_token = gr.Textbox(label="Hub Token", interactive=True, type="password")
         gr.Markdown("---")
             
-        with gr.Row():
-            gr.Markdown("### Ollama options")
-            with gr.Column():
-                ollama_create_local = gr.Checkbox(label="Create in Ollama (local)", value=False, interactive=True)
-                ollama_push_to_hub = gr.Checkbox(label="Push to Ollama", value=False, interactive=True)
-            with gr.Column():
-                ollama_model_name = gr.Textbox(label="Ollama Model Name", value="user/model_name")
-                ollama_pub_key = gr.Button("Ollama Pub Key")    
-        gr.Markdown("---")
+        # with gr.Row():
+        #     gr.Markdown("### Ollama options")
+        #     with gr.Column():
+        #         ollama_create_local = gr.Checkbox(label="Create in Ollama (local)", value=False, interactive=True)
+        #         ollama_push_to_hub = gr.Checkbox(label="Push to Ollama", value=False, interactive=True)
+        #     with gr.Column():
+        #         ollama_model_name = gr.Textbox(label="Ollama Model Name", value="user/model_name")
+        #         ollama_pub_key = gr.Button("Ollama Pub Key")    
+        save_output = gr.Markdown("---")
         save_button = gr.Button("Save Model", visible=True, interactive=True)
-        save_button.click(save_model, inputs=[model_name, hub_model_name, hub_token, gguf_16bit, gguf_8bit, gguf_4bit, gguf_custom, gguf_custom_value, merge_16bit, merge_4bit, just_lora, push_to_hub], outputs=[save_button])
+        save_button.click(save_model, inputs=[model_name, hub_model_name, hub_token, gguf_16bit, gguf_8bit, gguf_4bit, gguf_custom, gguf_custom_value, merge_16bit, merge_4bit, just_lora, push_to_hub], outputs=[save_output, save_button])
 
     with gr.Tab("Inference"):
         with gr.Row():