Hugging Face's logo

Spaces:
Lingo-IITGN
/
ganga-1b
Running

App Files Community
Lingo-IITGN commited on
Commit
2e23e90
·
verified ·
1 Parent(s): 14c4005

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -3,7 +3,7 @@ from transformers import AutoModelForCausalLM, AutoTokenizer, pipeline
3
 
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("LingoIITGN/ganga-1b")
6
- model = AutoModelForCausalLM.from_pretrained("LingoIITGN/ganga-1b", device_map="auto")
7
 
8
  @spaces.GPU(duration=120)
9
  def greet(input_text):
 
3
 
4
 
5
  tokenizer = AutoTokenizer.from_pretrained("LingoIITGN/ganga-1b")
6
+ model = AutoModelForCausalLM.from_pretrained("LingoIITGN/ganga-1b")
7
 
8
  @spaces.GPU(duration=120)
9
  def greet(input_text):