varun500 commited on
Commit
ae31f7b
1 Parent(s): dab5b1b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -7
app.py CHANGED
@@ -1,12 +1,19 @@
1
- import streamlit as st
2
- from transformers import pipeline, AutoTokenizer, AutoModelForQuestionAnswering
 
 
 
 
 
 
 
 
3
 
4
- # Load Flan Alpaca Large model
5
- model_name = "declare-lab/flan-alpaca-base"
6
- model = AutoModelForQuestionAnswering.from_pretrained(model_name)
7
- tokenizer = AutoTokenizer.from_pretrained(model_name)
8
- qa_pipeline = pipeline("question-answering", model=model, tokenizer=tokenizer)
9
 
 
 
 
10
  def main():
11
  # Set app title
12
  st.title("Flan Alpaca Large Model")
 
1
+ mport streamlit as st
2
+ from transformers import AutoTokenizer, AutoModelForCausalLM
3
+ import torch
4
+ import numpy as np
5
+ from transformers import (
6
+ AutoModelForCausalLM,
7
+ AutoTokenizer,
8
+ PreTrainedModel,
9
+ PreTrainedTokenizer
10
+ )
11
 
12
+ from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
 
 
 
13
 
14
+ tokenizer = AutoTokenizer.from_pretrained("declare-lab/flan-alpaca-large")
15
+
16
+ model = AutoModelForSeq2SeqLM.from_pretrained("declare-lab/flan-alpaca-large")
17
  def main():
18
  # Set app title
19
  st.title("Flan Alpaca Large Model")