Update nano_gpt_inferencing.py

nano_gpt_inferencing.py

@@ -193,5 +193,12 @@ def generate_paragraph(initial_text,max_token=50):
     final_model = model.to(device)
     encoded_text= encode(initial_text)
     encoded_text_tensor = torch.tensor(encoded_text).view(1, -1)
-    return decode(final_model.generate(encoded_text_tensor, max_new_tokens=int(max_token))[0].tolist())
+
+    try:
+        max_token = int(max_token)  
+        return decode(final_model.generate(encoded_text_tensor, max_new_tokens=max_token)[0].tolist())
+        
+    except ValueError as e:
+        return 'You have entered an invalid value in Max number of tokens, Please enter an integer value to proceed'
+