Hugging Face's logo

Spaces:
Lmcaboy
/
CS376Final
Sleeping

App Files Community
1
Lillian McAboy commited on
Commit
6ce8a67
·
1 Parent(s): ec363dc

work

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -1,4 +1,6 @@
1
  import streamlit as st
 
 
2
 
3
  model = 'google/gemma-1.1-7b-it'
4
 
 
1
  import streamlit as st
2
+ import torch, os
3
+ from transformers import AutoTokenizer, AutoModelForCausalLM, TextStreamer
4
 
5
  model = 'google/gemma-1.1-7b-it'
6