rename

app.py

@@ -34,15 +34,15 @@ Analyze the following IDA Hex Rays pseudocode and generate a valid JSON object c
 """
 
 @spaces.GPU
-def greet(name):
+def predict(code):
     # Prepare messages with system prompt
     messages = [
         {"role": "system", "content": system},
         {"role": "user", "content": "Hello, how are you?"}
     ]
     print(f"Test Tokenize: {tokenizer.apply_chat_template(messages, add_generation_prompt=True, tokenize=False)}")
-    return pipe(name)
+    return pipe(code)
 
-demo = gr.Interface(fn=greet, inputs="text", outputs="text",
+demo = gr.Interface(fn=predict, inputs="text", outputs="text",
                     description=frontmatter.load("README.md").content)
 demo.launch()