Update app.py

app.py

@@ -9,7 +9,10 @@ model = AutoModelForCausalLM.from_pretrained(model_path, torch_dtype=torch.bfloa
 
 # Define the inference function
 def generate_response(input_text, temperature, top_k, top_p):
-    inputs = tokenizer(input_text, return_tensors="pt")
+    before = f"# This is the assembly code:\n"#prompt
+    after = "\n# What is the source code?\n"#prompt
+    input_func = before+input_text.strip()+after
+    inputs = tokenizer(input_func, return_tensors="pt")
     outputs = model.generate(
         **inputs,
         max_length=512,  # Adjust this if needed