Update app.py

app.py

@@ -16,9 +16,11 @@ tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
 model = PeftModel.from_pretrained(model, peft_model_id)
 
 
-def make_inference(product, description):
+def make_inference(question: str)->str:
     batch = tokenizer(
-        f"Below is a product and description, please write a marketing email for this product.\n\n### Product:\n{product}\n### Description:\n{description}\n\n### Marketing Email",
+        "Below is a question, please write an answer for this question.\n\n"
+        f"### Question:\n{question}\n\n### Answer:\n",
+        #f"Below is a product and description, please write a marketing email for this product.\n\n### Product:\n{product}\n### Description:\n{description}\n\n### Marketing Email",
         return_tensors="pt",
     )
 
@@ -35,10 +37,10 @@ if __name__ == "__main__":
     gr.Interface(
         make_inference,
         [
-            gr.inputs.Textbox(lines=2, label="Product Name"),
-            gr.inputs.Textbox(lines=5, label="Product Description"),
+            gr.inputs.Textbox(lines=2, label="Medical Condition"),
+            #gr.inputs.Textbox(lines=5, label="Medical Condition Description"),
         ],
-        gr.outputs.Textbox(label="Ad"),
-        title="MarketMail-AI",
-        description="MarketMail-AI is a tool that generates marketing emails for products.",
+        gr.outputs.Textbox(label="Medical"),
+        title="MedChatBot",
+        description="MedChatBot is a tool that helps physicians with confidence in cancer diagnosis",
     ).launch()