Spaces:

LymphSteamer
/

low_haiku_space

Running

LymphSteamer commited on May 27

Commit

02cfa98

verified ·

1 Parent(s): 0d51f33

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,5 +1,6 @@
 import gradio as gr
 from transformers import AutoTokenizer,AutoModelForCasualLM
 def haiku_generate():
     model_id = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
     tokenizer = AutoTokenizer.from_pretrained(model_id)

 import gradio as gr
 from transformers import AutoTokenizer,AutoModelForCasualLM
+import torch
 def haiku_generate():
     model_id = "TinyLlama/TinyLlama-1.1B-Chat-v1.0"
     tokenizer = AutoTokenizer.from_pretrained(model_id)