NamuTechnology commited on
Commit
66cc82f
·
verified ·
1 Parent(s): 2d227ef

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +20 -1
app.py CHANGED
@@ -1,10 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  import gradio as gr
2
  from transformers import AutoModelForCausalLM, AutoTokenizer
3
  import torch
4
 
5
  # 모델 로딩
6
  print("모델 로딩 중...")
7
- model_name = "NamuTechnology/NamuLM"
8
 
9
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
10
  model = AutoModelForCausalLM.from_pretrained(
 
1
+ import os
2
+ import subprocess
3
+ import sys
4
+
5
+ # 필요한 패키지 자동 설치
6
+ def install_packages():
7
+ packages = [
8
+ "transformers==4.45.0",
9
+ "torch",
10
+ "accelerate",
11
+ "sentencepiece"
12
+ ]
13
+ for package in packages:
14
+ subprocess.check_call([sys.executable, "-m", "pip", "install", "-q", package])
15
+
16
+ print("필요한 패키지 설치 중...")
17
+ install_packages()
18
+ print("패키지 설치 완료!")
19
+
20
  import gradio as gr
21
  from transformers import AutoModelForCausalLM, AutoTokenizer
22
  import torch
23
 
24
  # 모델 로딩
25
  print("모델 로딩 중...")
26
+ model_name = "microsoft/Phi-3-mini-4k-instruct"
27
 
28
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
29
  model = AutoModelForCausalLM.from_pretrained(