cakemus commited on
Commit
6eb3422
·
1 Parent(s): 36bfcb2
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -6,7 +6,7 @@ from spaces import GPU # Import the GPU decorator for ZeroGPU
6
  @GPU
7
  def generate_text(prompt):
8
  # Load the model within the function so that it only runs on GPU when the function is called
9
- model = pipeline("text-generation", model="gpt2", device=0)
10
  return model(prompt, max_length=50)[0]["generated_text"]
11
 
12
  # Create the Gradio interface
 
6
  @GPU
7
  def generate_text(prompt):
8
  # Load the model within the function so that it only runs on GPU when the function is called
9
+ model = pipeline("text-generation", model="gpt3", device=0)
10
  return model(prompt, max_length=50)[0]["generated_text"]
11
 
12
  # Create the Gradio interface