Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -9,7 +9,8 @@ load_dotenv()
|
|
9 |
# Hugging Face API 정보
|
10 |
API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
|
11 |
API_KEY = os.getenv("HUGGINGFACE_API_KEY")
|
12 |
-
|
|
|
13 |
|
14 |
# 모델 호출 함수
|
15 |
def query_model(prompt):
|
@@ -18,8 +19,11 @@ def query_model(prompt):
|
|
18 |
|
19 |
response = requests.post(API_URL, headers=headers, json=data)
|
20 |
if response.status_code == 200:
|
|
|
|
|
21 |
return response.json().get("generated_text", "No output generated")
|
22 |
else:
|
|
|
23 |
return f"Error: {response.status_code}, {response.text}"
|
24 |
|
25 |
# Streamlit UI 구성
|
|
|
9 |
# Hugging Face API 정보
|
10 |
API_URL = "https://api-inference.huggingface.co/models/meta-llama/Meta-Llama-3-8B-Instruct"
|
11 |
API_KEY = os.getenv("HUGGINGFACE_API_KEY")
|
12 |
+
print(os.getenv("HUGGINGFACE_API_KEY"))
|
13 |
+
|
14 |
|
15 |
# 모델 호출 함수
|
16 |
def query_model(prompt):
|
|
|
19 |
|
20 |
response = requests.post(API_URL, headers=headers, json=data)
|
21 |
if response.status_code == 200:
|
22 |
+
result = response.json()
|
23 |
+
print("Response:", result)
|
24 |
return response.json().get("generated_text", "No output generated")
|
25 |
else:
|
26 |
+
print(f"Error: {response.status_code}, {response.text}")
|
27 |
return f"Error: {response.status_code}, {response.text}"
|
28 |
|
29 |
# Streamlit UI 구성
|