cweigendev commited on
Commit
89d0079
·
verified ·
1 Parent(s): 91d9b86

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,8 +2,8 @@ import gradio as gr
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoProcessor
4
 
5
- # Model name
6
- model_name = "DAMO-NLP-SG/VideoRefer-VideoLLaMA3-7B"
7
 
8
  # Global variables for model and processor
9
  model = None
 
2
  import torch
3
  from transformers import AutoModelForCausalLM, AutoProcessor
4
 
5
+ # Model name - USE THE BASE MODEL THAT HAS ALL FILES
6
+ model_name = "DAMO-NLP-SG/VideoLLaMA3-7B" # This has all the config files
7
 
8
  # Global variables for model and processor
9
  model = None