Jintonic92 commited on
Commit
59f9aef
·
verified ·
1 Parent(s): ca5a768

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -1
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
- st.write(f"Loaded API Key: {API_KEY}")
 
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 구성