Spaces:
Sleeping
Sleeping
BusinessDev
commited on
Commit
•
30b56f5
1
Parent(s):
dbbc1db
Update app.py
Browse files
app.py
CHANGED
@@ -2,7 +2,7 @@ import gradio as gr
|
|
2 |
from transformers import pipeline
|
3 |
import requests
|
4 |
|
5 |
-
model_id = "
|
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
|
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):
|