v3 of app

app.py

@@ -24,22 +24,29 @@ gemma_lm = GemmaCausalLM.from_preset("hf://bhaveshgoel07/MedCode")
 
 # Gradio app function
 def generate_response(prompt):
-    prompt = "you are a medical coder who's job is to help medical coders with ICD 10 codes"
     template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
     formatted_prompt = template.format(instruction=prompt, response="")
 
     # Generate output
-    output = gemma_lm.generate(prompt+formatted_prompt, max_length=100)
+    output = gemma_lm.generate(formatted_prompt, max_length=100)
 
     return output
 
-# Gradio interface
-interface = gr.Interface(
-    fn=generate_response,
-    inputs=gr.Textbox(label="Enter your prompt"),
-    outputs=gr.Textbox(label="Response"),
-    title="Fine-tuned GemmaCausalLM Chatbot",
-    description="Enter a prompt to get a response from the fine-tuned GemmaCausalLM model.",
-)
 
+# Use Gradio Blocks for more layout control
+with gr.Blocks() as interface:
+    gr.Markdown("## Fine-tuned GemmaCausalLM Chatbot")
+    gr.Markdown("Enter a prompt to get a response from the fine-tuned GemmaCausalLM model.")
+
+    with gr.Row():
+        prompt_input = gr.Textbox(label="Enter your prompt")
+        response_output = gr.Textbox(label="Response")
+
+    generate_button = gr.Button("Generate")
+    generate_button.click(generate_response, inputs=prompt_input, outputs=response_output)
+
+    # Add footer text at the bottom
+    gr.Markdown("Some Examples:\n 1. Cholera due to Vibrio cholerae 01, biovar cholerae ICD Code:A00.0\n 2. Cholera, Unspecified ICD Code:A00.9\n 3. Typhoid pneumonia, Typhoid Fever ICD Code:A01.03")
+
+# Launch the interface
 interface.launch()