Update app.py

app.py

@@ -1,18 +1,14 @@
 import gradio as gr
-from transformers import LlamaForConditionalGeneration, LlamaTokenizer
+from transformers import T5ForConditionalGeneration, T5Tokenizer
 
-# 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)
+model = T5ForConditionalGeneration.from_pretrained("t5-base")
+tokenizer = T5Tokenizer.from_pretrained("t5-base")
 
 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(