Spaces:

shrivasc
/

gpt2FineTuned

Build error

shrivasc commited on Feb 2, 2025

Commit

35f9a6d

verified ·

1 Parent(s): bb8ba55

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import AutoModelWithLMHead, AutoTokenizer
 # Load your model and tokenizer from the hub
 model = AutoModelWithLMHead.from_pretrained("shrivasc/my-finetuned-gpt2-medquad")  # Replace with your model path

 import gradio as gr
 from transformers import AutoModelWithLMHead, AutoTokenizer
+import torch
 # Load your model and tokenizer from the hub
 model = AutoModelWithLMHead.from_pretrained("shrivasc/my-finetuned-gpt2-medquad")  # Replace with your model path