Hugging Face's logo

Spaces:
PhysicsWallahAI
/
Aryabhata-Demo
Running

App Files Community
pw-ai-research commited on
Commit
a2cc02d
·
verified ·
1 Parent(s): 00f9a4a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import threading
9
 
10
  model_id = "PhysicsWallahAI/Aryabhata-1.0"
11
  tokenizer = AutoTokenizer.from_pretrained(model_id)
12
- model = AutoModelForCausalLM.from_pretrained(model_id, device_map="auto")
13
 
14
  def process_questions(example):
15
  example["question_text"] = example["question"]
 
9
 
10
  model_id = "PhysicsWallahAI/Aryabhata-1.0"
11
  tokenizer = AutoTokenizer.from_pretrained(model_id)
12
+ model = AutoModelForCausalLM.from_pretrained(model_id)
13
 
14
  def process_questions(example):
15
  example["question_text"] = example["question"]