acecalisto3 commited on
Commit
4639813
·
verified ·
1 Parent(s): 5ba31a8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -0
app.py CHANGED
@@ -48,6 +48,11 @@ TOOLS_DB.parent.mkdir(exist_ok=True)
48
 
49
  # Hugging Face Model
50
  model_name = "gpt2" # Use GPT-2 model for text generation
 
 
 
 
 
51
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
52
  generator = pipeline("text-generation", model=model_name, tokenizer=tokenizer, device=-1) # Use CPU
53
 
 
48
 
49
  # Hugging Face Model
50
  model_name = "gpt2" # Use GPT-2 model for text generation
51
+ # Hugging Face Model
52
+ model_name = "gpt2" # Ensure the model name is correct
53
+ if model_name is None:
54
+ raise ValueError("Model name is not specified. Please provide a valid model name.")
55
+
56
  tokenizer = GPT2Tokenizer.from_pretrained(model_name)
57
  generator = pipeline("text-generation", model=model_name, tokenizer=tokenizer, device=-1) # Use CPU
58