Update app.py

app.py

@@ -13,6 +13,7 @@ from pydantic import BaseModel
 hf_login(token=os.getenv("HF_TOKEN"))
 
 model_name = "meta-llama/Llama-3.2-1B-Instruct"
+model_name = "gregorlied/Llama-3.2-1B-Instruct-Medical-Report-Summarization"
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
 
 model = AutoModelForCausalLM.from_pretrained(
@@ -134,7 +135,6 @@ with gr.Blocks() as demo:
     with gr.Row():
         gr.CheckboxGroup(
             label="General Information", 
-            info="Which information do you want to include?",
             choices=[
                 "Age",
                 "Gender",
@@ -142,21 +142,15 @@ with gr.Blocks() as demo:
                 "Social History",
             ]
         )
-
-    with gr.Row():
         gr.CheckboxGroup(
             label="Medical History", 
-            info="Which information do you want to include?",
             choices=[
                 "Family History", 
                 "Medical Surgical History",
             ]
         )
-
-    with gr.Row():
         gr.CheckboxGroup(
             label="Signs and Symptoms", 
-            info="Which information do you want to include?",
             choices=[
                 "Signs and Symptoms",
                 "Comorbidities",
@@ -166,7 +160,6 @@ with gr.Blocks() as demo:
     with gr.Row():
         gr.CheckboxGroup(
             label="Medical Assesments", 
-            info="Which information do you want to include?",
             choices=[
                 "Diagnostic Techniques and Procedures",
                 "Laboratory Values",
@@ -174,11 +167,8 @@ with gr.Blocks() as demo:
                 "Diagnosis",                
             ]
         )
-
-    with gr.Row():
         gr.CheckboxGroup(
             label="Therapy and Results", 
-            info="Which information do you want to include?",
             choices=[
                 "Pharmacological Therapy",
                 "Interventional Therapy",