Update app.py

app.py

@@ -11,6 +11,12 @@ from transformers import pipeline
 client = InferenceClient("HuggingFaceH4/zephyr-7b-beta")
 #client = InferenceClient("google/gemma-1.1-7b-it")
 
+def clean_string(input_string):
+
+    characters_to_remove = [',', '/', ':', '"',';','|','-','_']
+    cleaned_string = ''.join(char for char in input_string if char not in characters_to_remove)
+    
+    return cleaned_string
 
 @spaces.GPU
 def respond(
@@ -49,6 +55,8 @@ def respond(
 prompt = """
 You are a cybersecurity analyst skilled in interpreting various types of logs resulting from different cyberattacks, including phishing, malware, advanced persistent threats, denial-of-service (DoS) and distributed denial-of-service (DDoS) attacks, man-in-the-middle (MitM) attacks, SQL injection attacks, and zero-day exploits. Using the following types of logs—login failures, event logs, firewall logs, and brute force logs—analyze the data and provide an interpretation of the analysis in English.
 
+**Important:** Do not use any information outside of the input provided. Focus solely on the data and indicators given in this prompt.
+
 Indicators to investigate:
 
 1. Multiple IP addresses signing in to the same account within a short period (e.g., 5 or more IPs in one hour).
@@ -68,7 +76,7 @@ Types of Indicators of Compromise (IOC) to consider:
 - Network Indicators: Irregular network traffic or connection patterns.
 - Behavioral Indicators: Activities that deviate from normal patterns.
 
-Response: Provide a detailed analysis and interpretation of the observed logs, focusing on identifying and explaining any potential security threats or breaches.
+Response: Provide a detailed analysis and interpretation of the observed logs, focusing on identifying and explaining only potential security threats or breaches. Ensure that your analysis is based solely on the information and indicators provided in this prompt.
 """
 
 demo = gr.ChatInterface(