SrijitMukherjee commited on
Commit
71d75bb
·
verified ·
1 Parent(s): 7c8e25d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -3,8 +3,9 @@ import streamlit as st
3
  # import ollama
4
  import transformers
5
  import torch
 
6
 
7
- model_id = "meta-llama/Meta-Llama-3-8B"
8
  #model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
9
 
10
  pipeline = transformers.pipeline(
 
3
  # import ollama
4
  import transformers
5
  import torch
6
+ from transformers import AutoModelForCausalLM, AutoTokenizer
7
 
8
+ model_id = "meta-llama/Llama-2-70b-chat"
9
  #model_id = "meta-llama/Meta-Llama-3-8B-Instruct"
10
 
11
  pipeline = transformers.pipeline(