Spaces:

SeemG
/

TheGiggleGrid

Paused

SeemG commited on Jul 13, 2024

Commit

35694d2

verified ·

1 Parent(s): 266231c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,7 @@ model2.load_state_dict(torch.load('gpt_124M_1.pth', map_location=torch.device(de
 # Define generation function
 def generate_text(prompt, max_length=50, num_return_sequences=10):
-    """ Generates humorous text using the GPT-2 model based on the provided prompt and user-specified parameters.
     Args:
         prompt (str): The starting text for the generation.

 # Define generation function
 def generate_text(prompt, max_length=50, num_return_sequences=10):
+    """ Text using the GPT-2 model based on the provided prompt and user-specified parameters.
     Args:
         prompt (str): The starting text for the generation.