Update fake.py

fake.py

@@ -2,14 +2,50 @@ import gradio as gr
 import spaces
 import transformers_gradio
 
-demo = gr.load(name="google/gemma-3-1b-it", src=transformers_gradio.registry)
+# Load the model interface
 demo = gr.load(name="unsloth/gemma-3-1b-it-GGUF", src=transformers_gradio.registry)
 
+# -------------------------------
+# 1. Custom doctor system prompt
+# -------------------------------
+DOCTOR_SYSTEM_PROMPT = """
+You are DoctorAI, a helpful, calm, and safe medical assistant.
+Give educational medical explanations. 
+Do NOT give diagnoses. 
+Do NOT prescribe treatments.
+Always advise seeing a medical professional for anything serious.
+"""
 
-demo.fn = spaces.GPU()(demo.fn)
 
+# ---------------------------------------------------
+# 2. Wrap original fn so doctor prompt is added
+# ---------------------------------------------------
+original_fn = demo.fn
+
+def doctor_wrapper(*args, **kwargs):
+    """
+    Intercepts user input and adds your custom doctor system prompt
+    before sending to the model.
+    """
+    # Usually args[0] is the user message in gradio chat interfaces
+    if len(args) > 0:
+        user_message = args[0]
+        new_message = DOCTOR_SYSTEM_PROMPT + "\n\nUser: " + user_message
+        args = (new_message,) + args[1:]
+
+    return original_fn(*args, **kwargs)
+
+
+# Replace the model fn with your wrapped version
+demo.fn = spaces.GPU()(doctor_wrapper)
+
+# Hide the API names
 for fn in demo.fns.values():
     fn.api_name = False
 
+
+# ---------------------------------------------------
+# 3. Launch
+# ---------------------------------------------------
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()