Update app.py

app.py

@@ -16,24 +16,35 @@ def generate_response(input_text, temperature, top_k, top_p):
     outputs = model.generate(
         **inputs,
         max_length=512,  # Adjust this if needed
-        do_sample=True,
-        top_k=int(top_k),
-        top_p=float(top_p),
-        temperature=float(temperature)
     )
+    #   do_sample=True,
+    #    top_k=int(top_k),
+    #    top_p=float(top_p),
+    #    temperature=float(temperature)
     response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return response
+    # Split the response into assembly and source code (if applicable)
+    if "# This is the assembly code:" in response:
+        parts = response.split("# What is the source code?")
+        assembly_code = parts[0].replace("# This is the assembly code:", "").strip()
+        source_code = parts[1].strip() if len(parts) > 1 else ""
+        return f"```c\n{assembly_code}\n```", f"```c\n{source_code}\n```"
+    else:
+        return "No assembly code found.", "No source code found."
 
 # Create a Gradio interface with sliders
 interface = gr.Interface(
     fn=generate_response,
     inputs=[
-        gr.Textbox(lines=5, placeholder="Enter your input text here...", label="Input Text"),
-        gr.Slider(0.1, 2.0, value=0.0, step=0.1, label="Temperature"),
-        gr.Slider(1, 100, value=10, step=1, label="Top-k"),
+        gr.Textbox(lines=5, placeholder="Enter assembly code here...", label="Input Assembly Code"),
+        gr.Slider(0.1, 2.0, value=0.7, step=0.1, label="Temperature"),
+        gr.Slider(1, 100, value=50, step=1, label="Top-k"),
         gr.Slider(0.1, 1.0, value=0.95, step=0.05, label="Top-p")
     ],
-    outputs=gr.Textbox(label="Generated Response"),
+    outputs=[
+        gr.Markdown(label="Assembly Code"),
+        gr.Markdown(label="Source Code")
+    ],
+
     title="Falcon decompiler Interactive Demo",
     description="Adjust the sliders for temperature, top-k, and top-p to customize the model's response."
 )