DIVY118 commited on
Commit
61dcbec
·
verified ·
1 Parent(s): 27c7f8a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -10
app.py CHANGED
@@ -78,22 +78,16 @@ def Web():
78
  }
79
  return jsonify(result)
80
 
81
- @app.route('/imageneration', methods=['POST'])
82
  def IMGEN():
83
- data = request.json
84
- prompt = data.get('prompt', '')
85
- key = data.get('key', '')
86
 
87
  API_URL = "https://api-inference.huggingface.co/models/OEvortex/HelpingAI-PixelCraft"
88
  headers = {"Authorization": f"Bearer {key}"}
89
-
90
- def query(payload):
91
- response = requests.post(API_URL, headers=headers, json=payload)
92
- return response.content
93
-
94
 
95
  result = {
96
- 'response': query({"inputs": prompt,})
97
  }
98
 
99
  return jsonify(result)
 
78
  }
79
  return jsonify(result)
80
 
81
+ @app.route('/imageneration', methods=['GET'])
82
  def IMGEN():
83
+ prompt = request.args.get('prompt', '')
84
+ key = request.args.get('key', '')
 
85
 
86
  API_URL = "https://api-inference.huggingface.co/models/OEvortex/HelpingAI-PixelCraft"
87
  headers = {"Authorization": f"Bearer {key}"}
 
 
 
 
 
88
 
89
  result = {
90
+ 'response': query({"inputs": requests.post(API_URL, headers=headers, json=payload).content})
91
  }
92
 
93
  return jsonify(result)