Update app.py
Browse files
app.py
CHANGED
|
@@ -67,6 +67,7 @@ import gradio as gr
|
|
| 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")
|
|
|
|
| 67 |
from transformers import AutoModelForCausalLM, AutoTokenizer
|
| 68 |
import torch
|
| 69 |
|
| 70 |
+
|
| 71 |
model_name = "Asit03/AI_Agent_V2"
|
| 72 |
tokenizer = AutoTokenizer.from_pretrained(model_name)
|
| 73 |
model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
|