Update app.py

app.py

@@ -23,8 +23,8 @@ model = AutoModelForCausalLM.from_pretrained(
     trust_remote_code=True,
 )
 
-checkpoint = "gregorlied/Llama-3.2-1B-Instruct-Medical-Report-Summarization"
-model = PeftModel.from_pretrained(model, checkpoint)
+# checkpoint = "gregorlied/Llama-3.2-1B-Instruct-Medical-Report-Summarization"
+# model = PeftModel.from_pretrained(model, checkpoint)
 # model = model.merge_and_unload()
 
 class Person(BaseModel):
@@ -107,9 +107,9 @@ def summarize(text):
         # top_p=0.9,
         # do_sample=True,
         # temperature=0.6,
-        min_new_tokens=50,
+        # min_new_tokens=50,
         max_new_tokens=2048,
-        # logits_processor=[xgr_logits_processor]
+        logits_processor=[xgr_logits_processor]
     )
 
     generated_ids = [