mou3az commited on
Commit
c42ba06
·
verified ·
1 Parent(s): 766a0de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +11 -26
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
- retries = 0
37
- MAX_RETRIES = 3
38
- response = None
39
-
40
- while retries < MAX_RETRIES:
41
- retries += 1
42
- try:
43
- response = client.text_generation(
44
- formatted_prompt,
45
- **generate_kwargs,
46
- stream=False,
47
- details=False,
48
- return_full_text=False,
49
- )
50
- break
51
 
52
- except (RequestException, SystemExit) as e:
53
- if retries == MAX_RETRIES:
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():