BusinessDev commited on
Commit
30b56f5
1 Parent(s): dbbc1db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
2
  from transformers import pipeline
3
  import requests
4
 
5
- model_id = "meta-llama/Meta-Llama-3-8B" # You can replace this with any model of your choice
6
 
7
  def fetch_s3_text_file(url):
8
  try:
@@ -13,9 +13,9 @@ def fetch_s3_text_file(url):
13
  print(f"Error fetching the file: {e}")
14
  return None
15
 
16
- access_token = fetch_s3_text_file("https://mybookbooks.s3.amazonaws.com/key.txt")
17
 
18
- generator = pipeline("text-generation", model=model_id, token = access_token)
19
 
20
  # Define the function to process the input and generate text
21
  def generate_text(prompt):
 
2
  from transformers import pipeline
3
  import requests
4
 
5
+ model_id = "bartowski/gemma-2-27b-it-GGUF" # You can replace this with any model of your choice
6
 
7
  def fetch_s3_text_file(url):
8
  try:
 
13
  print(f"Error fetching the file: {e}")
14
  return None
15
 
16
+ #access_token = fetch_s3_text_file("https://mybookbooks.s3.amazonaws.com/key.txt")
17
 
18
+ generator = pipeline("text-generation", model=model_id)
19
 
20
  # Define the function to process the input and generate text
21
  def generate_text(prompt):