Update app.py

app.py

@@ -3,7 +3,7 @@ import os
 import google.generativeai as genai
 import logging
 import time
-#import backoff
+import backoff
 
 # Configure Logging
 logging.basicConfig(level=logging.ERROR, format='%(asctime)s - %(levelname)s - %(message)s')
@@ -28,9 +28,9 @@ custom_css = """
 }
 """
 
-#@backoff.on_exception(backoff.expo,
-                      #(genai.APIError),
-                      #max_tries=3) # retry up to 3 times
+@backoff.on_exception(backoff.expo,
+                      (genai.APIError),
+                      max_tries=3) # retry up to 3 times
 def predict(prompt):
     # Create the model
     generation_config = {