Spaces:

chbsaikiran
/

NextWordGPT

Running

chbsaikiran commited on Jan 17

Commit

f9f8ce6

1 Parent(s): 49c48e3

Bug Fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import gradio as gr
 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")  # Using GPT-2 tokenizer for compatibility
 # Load model
-from train_get2_8_init import GPT, GPTConfig
 device = 'cuda' if torch.cuda.is_available() else 'cpu'

 tokenizer = GPT2Tokenizer.from_pretrained("gpt2")  # Using GPT-2 tokenizer for compatibility
 # Load model
+from train_get2-8-init import GPT, GPTConfig
 device = 'cuda' if torch.cuda.is_available() else 'cpu'