Back to beam search; add the correct prompt command

app.py

@@ -16,27 +16,22 @@ def paraphrase(source_text):
         headers={"Authorization": f"Bearer {os.environ['TOKEN']}"},
         data=json.dumps(
             {
-                "inputs": source_text,
+                "inputs": "paraphrase: " + source_text,
                 "parameters": {
                     # "do_sample": True,
-                    "num_beams": 10,
-                    "top_k": 5,
-                    "repetition_penalty": 2.0,
                     "temperature": 1.5,
+                    "top_k": 5,
+                    "num_beams": 10,
                     "num_return_sequences": 10,
-                    "max_length": 200,
+                    "max_length": 100,
                 },
             }
         ),
     )
     if not (res.status_code == 200):
-        raise ValueError(
-            "Could not complete request to HuggingFace API, Error {}".format(
-                res.status_code
-            )
-        )
+        return f"Got a {res.status_code} status code from HuggingFace."
     results = res.json()
-    print(results)
+    # print(results)
     outputs = [
         x["generated_text"]
         for x in results