Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
|