Create main.py

main.py

@@ -0,0 +1,27 @@
+import gradio as gr
+from transformers import LlamaForConditionalGeneration, LlamaTokenizer
+
+# Load pre-trained model and tokenizer
+model_name = "meta-ai/llama-3.2-3b-instruct"
+model = LlamaForConditionalGeneration.from_pretrained(model_name)
+tokenizer = LlamaTokenizer.from_pretrained(model_name)
+
+def evaluate_password_strength(password):
+    # Use the Llama model to evaluate the password strength
+    input_text = f"Rate the strength of the password: {password}"
+    inputs = tokenizer(input_text, return_tensors="pt")
+    output = model.generate(**inputs)
+    response = tokenizer.decode(output[0], skip_special_tokens=True)
+    
+    return response
+
+demo = gr.Interface(
+    evaluate_password_strength,
+    gr.Textbox(label="Enter your password"),
+    gr.Textbox(label="Password Strength Evaluation"),
+    title="Password Strength Evaluator",
+    description="Get the AI's evaluation of your password strength.",
+)
+
+if __name__ == "__main__":
+    demo.launch()