gpt3

app.py

@@ -6,7 +6,7 @@ from spaces import GPU  # Import the GPU decorator for ZeroGPU
 @GPU
 def generate_text(prompt):
     # Load the model within the function so that it only runs on GPU when the function is called
-    model = pipeline("text-generation", model="gpt2", device=0)
+    model = pipeline("text-generation", model="gpt3", device=0)
     return model(prompt, max_length=50)[0]["generated_text"]
 
 # Create the Gradio interface