Update app.py

app.py

@@ -1,11 +1,8 @@
 import gradio as gr
 from huggingface_hub import InferenceClient
 import os
-import pandas as pd
 from mitreattack.stix20 import MitreAttackData
-from format_input import detect_log_type
 from descriptions import descriptions
-from testing_input import log_input
 
 # Chemins des fichiers JSON
 ics_attack_path = 'ics-attack.json'
@@ -40,7 +37,7 @@ def respond(
             messages.append({"role": "assistant", "content": val[1]})
 
     messages.append({"role": "user", "content": message})
-    descriptions = descriptions(messages["content"])
+    message_content = message
 
     response = ""
 
@@ -61,13 +58,11 @@ demo = gr.ChatInterface(
     respond,
     additional_inputs=[
         gr.Textbox(value=f"""<s>[INST] Given these TTPs: {techniques_str}\n\n and here are {descriptions}\n\nfigure out which technique is used in these logs and respond in bullets points and nothing else[/INST]""", label="System message"),
-            gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens"),
+        gr.Slider(minimum=1, maximum=2048, value=1024, step=1, label="Max new tokens"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.1, step=0.1, label="Temperature"),
         gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p (nucleus sampling)"),
     ],
 )
 
-
-
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch()