gpt3
Browse files
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="
|
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
|