Mihaiii commited on
Commit
1a41eb3
·
verified ·
1 Parent(s): 5f21c0e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ import torch
12
  import gradio as gr
13
  from transformers import AutoModelForCausalLM, TextIteratorStreamer
14
 
15
- model_name = 'AIDC-AI/Ovis2-16B'
16
  use_thread = False
17
 
18
  # load model
 
12
  import gradio as gr
13
  from transformers import AutoModelForCausalLM, TextIteratorStreamer
14
 
15
+ model_name = 'AIDC-AI/Ovis2-2B'
16
  use_thread = False
17
 
18
  # load model