Update app.py

app.py

@@ -568,7 +568,7 @@ def summarize(
     prefix = "summarize: "
     ret = ""
 
-    for index in math.ceil( len(words) / 512 ):
+    for index in range(math.ceil( len(words) / 512 )):
         
         chunk = " ".join(words[ index*512:(index+1)*512 ])
         inputs = tokenizer.encode( prefix + chunk, return_tensors="pt", max_length=float('inf'), truncation=False)
@@ -722,7 +722,7 @@ def translate(txt,to_lang="en",from_lang=False):
     prefix = f"translate {language_codes[from_lang]} to {language_codes[to_lang]}: "
     words = txt.split()
     ret = ""
-    for index in math.ceil( len(words) / 512 ):
+    for index in range(math.ceil( len(words) / 512 )):
         chunk = " ".join(words[ index*512:(index+1)*512 ])
         inputs = tokenizer.encode(prefix+chunk, return_tensors="pt", max_length=float('inf'), truncation=False)
         gen = model.generate(chunk,input)