Update app.py
Browse files
app.py
CHANGED
@@ -1,6 +1,5 @@
|
|
1 |
from flask import Flask, render_template, request, jsonify
|
2 |
from huggingface_hub import InferenceClient
|
3 |
-
import time
|
4 |
from requests.exceptions import RequestException
|
5 |
|
6 |
app = Flask(__name__)
|
@@ -33,32 +32,18 @@ def generate_quiz(context):
|
|
33 |
do_sample=True,
|
34 |
seed=42,)
|
35 |
|
36 |
-
|
37 |
-
|
38 |
-
|
39 |
-
|
40 |
-
|
41 |
-
|
42 |
-
|
43 |
-
|
44 |
-
|
45 |
-
**generate_kwargs,
|
46 |
-
stream=False,
|
47 |
-
details=False,
|
48 |
-
return_full_text=False,
|
49 |
-
)
|
50 |
-
break
|
51 |
|
52 |
-
|
53 |
-
|
54 |
-
return {"error": f"Failed after {MAX_RETRIES} retries. Error: {str(e)}"}
|
55 |
-
|
56 |
-
time.sleep(retries)
|
57 |
-
|
58 |
-
if response is None:
|
59 |
-
return {"error": "Max retries exceeded without success."}
|
60 |
-
|
61 |
-
return response
|
62 |
|
63 |
@app.route("/", methods=["GET", "POST"])
|
64 |
def generate_quiz_page():
|
|
|
1 |
from flask import Flask, render_template, request, jsonify
|
2 |
from huggingface_hub import InferenceClient
|
|
|
3 |
from requests.exceptions import RequestException
|
4 |
|
5 |
app = Flask(__name__)
|
|
|
32 |
do_sample=True,
|
33 |
seed=42,)
|
34 |
|
35 |
+
try:
|
36 |
+
response = client.text_generation(
|
37 |
+
formatted_prompt,
|
38 |
+
**generate_kwargs,
|
39 |
+
stream=False,
|
40 |
+
details=False,
|
41 |
+
return_full_text=False,
|
42 |
+
)
|
43 |
+
return response
|
|
|
|
|
|
|
|
|
|
|
|
|
44 |
|
45 |
+
except (RequestException, SystemExit) as e:
|
46 |
+
return {"error": str(e)}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
47 |
|
48 |
@app.route("/", methods=["GET", "POST"])
|
49 |
def generate_quiz_page():
|