Jintonic92 commited on
Commit
99d03e3
·
verified ·
1 Parent(s): 5728374

Upload 3 files

Browse files
Files changed (2) hide show
  1. app.py +37 -0
  2. requirements.txt +14 -0
app.py ADDED
@@ -0,0 +1,37 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import requests
2
+ import streamlit as st
3
+ from dotenv import load_dotenv
4
+ import os
5
+
6
+ # .env 파일 로드
7
+ load_dotenv()
8
+
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
+ def query_model(prompt):
14
+ headers = {"Authorization": f"Bearer {API_KEY}"}
15
+ data = {"inputs": prompt}
16
+
17
+ response = requests.post(API_URL, headers=headers, json=data)
18
+ if response.status_code == 200:
19
+ return response.json().get("generated_text", "No output generated")
20
+ else:
21
+ return f"Error: {response.status_code}, {response.text}"
22
+
23
+ # Streamlit UI 구성
24
+ st.title("Meta-Llama Text Generator")
25
+ st.write("Enter a prompt to generate text using the Meta-Llama-3B model.")
26
+
27
+ # 사용자 입력
28
+ prompt = st.text_area("Enter your prompt:", height=200)
29
+
30
+ if st.button("Generate"):
31
+ if prompt.strip():
32
+ st.write("Generating...")
33
+ output = query_model(prompt)
34
+ st.write("### Output:")
35
+ st.write(output)
36
+ else:
37
+ st.warning("Please enter a valid prompt!")
requirements.txt ADDED
@@ -0,0 +1,14 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ streamlit>=1.28.0
2
+ pandas>=2.1.0
3
+ Pillow>=10.0.0
4
+ numpy>=1.24.0
5
+ protobuf>=4.21.0
6
+ watchdog>=3.0.0
7
+ python-dotenv==1.0.1
8
+ requests
9
+
10
+
11
+ tqdm>=4.65.0
12
+ transformers>=4.30.0
13
+ sentence-transformers>=2.2.2
14
+ scikit-learn>=1.3.0