Spaces:

pro-grammer
/

StoryCrafterLLM

Running

pro-grammer commited on Oct 1, 2024

Commit

8154e9b

verified ·

1 Parent(s): 3fc3d9d

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -3,7 +3,7 @@ import torch
 import tiktoken
 import gradio as gr
 from transformers import GPT2Tokenizer
-from src.gpt import GPTLanguageModel  # Assuming the model is stored in src/gpt.py
 # Initialize the GPT-2 tokenizer
 enc = tiktoken.get_encoding("gpt2")  # Using tiktoken

 import tiktoken
 import gradio as gr
 from transformers import GPT2Tokenizer
+from model import GPTLanguageModel
 # Initialize the GPT-2 tokenizer
 enc = tiktoken.get_encoding("gpt2")  # Using tiktoken