Hugging Face's logo

Spaces:
cpv2280
/
gpt2-tinystories-generator
Sleeping

App Files Community
cpv2280 commited on
Commit
9845e29
·
verified ·
1 Parent(s): 8ae6c4b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +0 -2
app.py CHANGED
@@ -27,8 +27,6 @@ from sentence_transformers import SentenceTransformer, util
27
  from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
28
 
29
 
30
-
31
-
32
  device = "cuda" if torch.cuda.is_available() else "cpu" # ✅ Use GPU if available
33
 
34
  # Load fine-tuned GPT-2 model
 
27
  from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
28
 
29
 
 
 
30
  device = "cuda" if torch.cuda.is_available() else "cpu" # ✅ Use GPU if available
31
 
32
  # Load fine-tuned GPT-2 model