Spaces:
Runtime error
Runtime error
Commit History
-
Update requirements.txt
Verified signaturea68a1af -
app.py
Verified signatureimport streamlit as st from transformers import AutoTokenizer, AutoModelForCausalLM import torch # Load the Phi 2 model and tokenizer tokenizer = AutoTokenizer.from_pretrained("microsoft/phi-2", trust_remote_code=True) model = AutoModelForCausalLM.from_pretrained("microsoft/phi-2", device_map="auto", trust_remote_code=True) # Streamlit UI st.title("Microsoft Phi 2 Streamlit App") prompt = st.text_area("Enter your prompt", "Give me a list of 13 words that have 9 letters.") if st.button("Generate Output"): # Code for generating output based on user input with torch.no_grad(): token_ids = tokenizer.encode(prompt, add_special_tokens=False, return_tensors="pt") output_ids = model.generate( token_ids.to(model.device), max_new_tokens=512, do_sample=True, temperature=0.3 ) output = tokenizer.decode(output_ids[0][token_ids.size(1):]) st.text("Generated Output:") st.write(output)aee6203 -
requirements.txt
Verified signaturetransformers torchfd39b97 -
initial commit
Verified signature06bfc4e