Update app.py

app.py

@@ -2,7 +2,7 @@
 # app.py
 
 import gradio as gr
-from transformers import AutoTokenizer
+from transformers import AutoTokenizer, AutoModelForCausalLM
 from groq import Groq
 import os
 from huggingface_hub import login
@@ -22,8 +22,16 @@ if hf_token is None:
 login(token=hf_token)
 
 # Model identifier for Groq API (you can replace it with your HF model if needed)
+# Model identifier
 model_name = "asthaa30/l3.1"
 
+# Load tokenizer and model directly
+try:
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+except Exception as e:
+    raise ValueError(f"Failed to load model or tokenizer: {e}")
+
 # Load tokenizer (model will be accessed via Groq API)
 try:
     tokenizer = AutoTokenizer.from_pretrained(model_name, token=hf_token)