feat: a

app.py

@@ -7,11 +7,11 @@ def greet(name):
 
 
 @spaces.GPU
-def infer():
+def infer(name: str = "Who are you?"):
     from transformers import pipeline
 
     messages = [
-        {"role": "user", "content": "Who are you?"},
+        {"role": "user", "content": name},
     ]
     pipe = pipeline("text-generation", model="microsoft/Phi-3-mini-4k-instruct", trust_remote_code=True)
     return pipe(messages)