Update app.py

app.py

@@ -1,17 +1,11 @@
 import gradio as gr
-from transformers import AutoConfig, AutoModelForCausalLM
+from transformers import AutoTokenizer, AutoModelForCausalLM
 from modeling_snowflake import Snowflake4CausalLM
 import torch
 
-# --- Load Model and Tokenizer from Hugging Face Hub ---
-# Register the custom model with the transformers library
-AutoConfig.register("SnowflakeCore", PretrainedConfig)
-AutoModelForCausalLM.register(PretrainedConfig, Snowflake4CausalLM)
-
-# Load tokenizer
+# Load tokenizer and model
+MODEL_NAME = "FlameF0X/Snowflake-G0-stable"
 tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-
-# Load model
 model = AutoModelForCausalLM.from_pretrained(
     MODEL_NAME,
     torch_dtype=torch.float16  # Use half precision for memory efficiency