Spaces:

Asit03
/

Bot

Sleeping

Asit03 commited on Apr 30, 2025

Commit

3a90a32

verified ·

1 Parent(s): 62a6ccd

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -70,7 +70,7 @@ import torch
 model_name = "Asit03/AI_Agent_V2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForCausalLM.from_pretrained(model_name, device_map="auto")
 def generate_response(prompt):
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)

 model_name = "Asit03/AI_Agent_V2"
 tokenizer = AutoTokenizer.from_pretrained(model_name)
+model = AutoModelForCausalLM.from_pretrained(model_name)
 def generate_response(prompt):
     inputs = tokenizer(prompt, return_tensors="pt").to(model.device)