neo125m

app.py

@@ -6,7 +6,8 @@ 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="gpt3", device=0)
+    model = pipeline("text-generation", model="EleutherAI/gpt-neo-125M", device=0)
+
     return model(prompt, max_length=50)[0]["generated_text"]
 
 # Create the Gradio interface