Update README.md

README.md

@@ -22,7 +22,7 @@ from transformers import pipeline
 # Load text-generation pipeline
 generator = pipeline(
     "text-generation",
-    model="TanishkB/RandomNumber",
+    model="TanishkB/RandomNumberGenerator",
     device=-1   # use 0 if you have GPU
 )
 
@@ -39,7 +39,7 @@ while True:
     # Generate reply
     response = generator(
         prompt,
-        max_new_tokens=128,
+        max_new_tokens=64,
         return_full_text=False
     )[0]["generated_text"]