1MR commited on
Commit
7fa2d88
·
verified ·
1 Parent(s): 7c7b36a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -1,5 +1,11 @@
1
- # Load model directly
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
 
 
4
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-4B")
5
- model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-4B")
 
 
 
 
 
1
+ from fastapi import FastAPI
2
  from transformers import AutoTokenizer, AutoModelForCausalLM
3
 
4
+ app = FastAPI()
5
+
6
  tokenizer = AutoTokenizer.from_pretrained("Qwen/Qwen3-4B")
7
+ model = AutoModelForCausalLM.from_pretrained("Qwen/Qwen3-4B")
8
+
9
+ @app.get("/")
10
+ def read_root():
11
+ return {"message": "Model is ready"}