Add updated app file

app.py

@@ -1,11 +1,22 @@
+from transformers import PretrainedConfig, AutoModelForCausalLM, AutoTokenizer
 import streamlit as st
-import torch
-from transformers import AutoModelForCausalLM, AutoTokenizer, AutoConfig
 
-# Load the configuration from the Hugging Face repo
-config = AutoConfig.from_pretrained("praty7717/Odeyssey")
+# Define your custom config class
+class MyCustomConfig(PretrainedConfig):
+    model_type = "gpt"  # Specify the model type
 
-# Load the model and tokenizer using the config
+    def __init__(self, vocab_size, n_embd, n_layer, n_head, block_size, **kwargs):
+        super().__init__(**kwargs)
+        self.vocab_size = vocab_size
+        self.n_embd = n_embd
+        self.n_layer = n_layer
+        self.n_head = n_head
+        self.block_size = block_size
+
+# Load your model configuration
+config = MyCustomConfig.from_json_file("config.json")  # Make sure to specify the correct path
+
+# Load the model and tokenizer
 model = AutoModelForCausalLM.from_pretrained("praty7717/Odeyssey", config=config)
 tokenizer = AutoTokenizer.from_pretrained("praty7717/Odeyssey")