Hugging Face's logo

Spaces:
aryo100
/
qwen_api
Sleeping

App Files Community
aryo100 commited on
Commit
46ce46c
·
1 Parent(s): dfe0fb0

update model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import torch
5
  app = FastAPI()
6
 
7
  # Load Qwen-7B dengan trust_remote_code
8
- model_name = "Qwen/Qwen-1.8B"
9
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
10
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu", trust_remote_code=True)
11
 
 
5
  app = FastAPI()
6
 
7
  # Load Qwen-7B dengan trust_remote_code
8
+ model_name = "Qwen/Qwen-1_8B"
9
  tokenizer = AutoTokenizer.from_pretrained(model_name, trust_remote_code=True)
10
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="cpu", trust_remote_code=True)
11