Hugging Face's logo

Spaces:
JuanMa360
/
DD360-Bot
Running

App Files Community
JuanMa360 commited on
Commit
b1e9dee
·
1 Parent(s): 99c5436

refactor: model

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from PIL import Image
4
  import gradio as gr
5
 
6
  model = AutoModelForCausalLM.from_pretrained(
7
- "dd360-v1-3b",
8
  torch_dtype=torch.float32,
9
  device_map="auto")
10
  tokenizer = AutoTokenizer.from_pretrained("tokenizer-dd360",torch_dtype=torch.float32)
 
4
  import gradio as gr
5
 
6
  model = AutoModelForCausalLM.from_pretrained(
7
+ "MILVLG/imp-v1-3b",
8
  torch_dtype=torch.float32,
9
  device_map="auto")
10
  tokenizer = AutoTokenizer.from_pretrained("tokenizer-dd360",torch_dtype=torch.float32)