Asit03 commited on
Commit
6f289cf
·
verified ·
1 Parent(s): e5bffdf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -67,7 +67,7 @@ import gradio as gr
67
  from transformers import AutoModelForCausalLM, AutoTokenizer
68
  import torch
69
 
70
- model_name = "Asit03/ChatSeek-8B-v2"
71
  tokenizer = AutoTokenizer.from_pretrained(model_name)
72
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
73
 
 
67
  from transformers import AutoModelForCausalLM, AutoTokenizer
68
  import torch
69
 
70
+ model_name = "Asit03/AI_Agent_V2"
71
  tokenizer = AutoTokenizer.from_pretrained(model_name)
72
  model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
73