Spaces:

Antoniskaraolis
/

Embeddings

Runtime error

Antoniskaraolis commited on Apr 14, 2024

Commit

693890a

verified ·

1 Parent(s): 9d3d729

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -23,7 +23,6 @@ def answer_question(model, tokenizer, question):
 tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
 tokenizer.pad_token = tokenizer.eos_token
 model = GPT2LMHeadModel.from_pretrained('distilgpt2')
-model.to('cuda')  # Make sure to use CUDA if available
 def setup_gradio_interface():
     with gr.Blocks() as demo:

 tokenizer = GPT2Tokenizer.from_pretrained('distilgpt2')
 tokenizer.pad_token = tokenizer.eos_token
 model = GPT2LMHeadModel.from_pretrained('distilgpt2')
 def setup_gradio_interface():
     with gr.Blocks() as demo: