Spaces:

FlameF0X
/

SnowflakeCore-Demo-Inteface

Sleeping

FlameF0X commited on Apr 26

Commit

5a44dc6

verified ·

1 Parent(s): c6c6cd8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -1,6 +1,7 @@
 import os
 import gradio as gr
 import torch
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import datetime

 import os
 import gradio as gr
 import torch
+from transformers import TextGenerationPipeline
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import datetime