sailormars18 commited on
Commit
f568152
·
1 Parent(s): bcb1530

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -8,6 +8,7 @@ import transformers
8
 
9
  # Define a function for generating text based on a prompt using the fine-tuned GPT-2 model and the tokenizer
10
  def generate_text(prompt, length=100, theme=None, **kwargs):
 
11
  model_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/pytorch_model.bin"
12
 
13
  # Load the model from the Hugging Face space
 
8
 
9
  # Define a function for generating text based on a prompt using the fine-tuned GPT-2 model and the tokenizer
10
  def generate_text(prompt, length=100, theme=None, **kwargs):
11
+
12
  model_url = "https://huggingface.co/spaces/sailormars18/Yelp-reviews-usingGPT2/blob/main/pytorch_model.bin"
13
 
14
  # Load the model from the Hugging Face space