Create app.py

app.py

@@ -0,0 +1,32 @@
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import gradio as gr
+
+# Load the model and tokenizer from Hugging Face Hub
+model_name = "blizet/Llama-Phishing-Finetune"
+
+# Load model and tokenizer
+tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
+
+# Define a function to generate responses
+def analyze_email(email):
+    prompt = f"""
+    Analyze the email with the following criteria:
+    ...
+    Email: {email}
+    """
+    inputs = tokenizer(prompt, return_tensors="pt", truncation=True)
+    outputs = model.generate(**inputs, max_new_tokens=150)
+    return tokenizer.decode(outputs[0], skip_special_tokens=True)
+
+# Gradio interface
+demo = gr.Interface(
+    fn=analyze_email,
+    inputs="textbox",
+    outputs="textbox",
+    title="Phishing Email Analyzer",
+    description="Enter an email for analysis."
+)
+
+# Launch the app
+demo.launch()