Update app.py

app.py

@@ -58,12 +58,15 @@ def generate_v3(inputs, count):
     return sentences
 
 
-def generate_v4(inputs, count):
+def generate_v4(encoding, count):
     """Generate text using a Diverse Beam Search strategy."""
-    input_ids, attention_masks = inputs["input_ids"], inputs["attention_mask"]
+    print(encoding)
+    input_ids, attention_masks = encoding["input_ids"], encoding["attention_mask"]
+    print(input_ids)
+    print(attention_masks)
     outputs = model.generate(input_ids=input_ids, 
                              attention_mask=attention_masks,
-                             max_length=512,  # 256,
+                             max_length=512,
                              do_sample=True,
                              top_k=120,
                              top_p=0.95,
@@ -109,10 +112,10 @@ def paraphrase(sentence: str, count: str):
     
     return {
         'result': {
-            'generate_v1':generate_v1(encoding, count),
-            'generate_v2':generate_v2(encoding, count),
-            'generate_v3':generate_v3(encoding, count),
-            'generate_v4':generate_v4(encoding, count),
+            # 'generate_v1':generate_v1(encoding, count),
+            # 'generate_v2':generate_v2(encoding, count),
+            # 'generate_v3':generate_v3(encoding, count),
+            # 'generate_v4':generate_v4(encoding, count),
         }
     }