Update main.py
Browse files
main.py
CHANGED
|
@@ -20,11 +20,7 @@ from linebot.models import (
|
|
| 20 |
# # Initialize the Gemini model
|
| 21 |
# model = genai.GenerativeModel('gemini-pro')
|
| 22 |
|
| 23 |
-
client = Client("https://alanchen1115-gemini-api.hf.space/--replicas/mgfss/")
|
| 24 |
-
result = client.predict(
|
| 25 |
-
"Hello!!", # str in 'Prompt' Textbox component
|
| 26 |
-
api_name="/predict"
|
| 27 |
-
)
|
| 28 |
|
| 29 |
app = FastAPI()
|
| 30 |
app.add_middleware(
|
|
|
|
| 20 |
# # Initialize the Gemini model
|
| 21 |
# model = genai.GenerativeModel('gemini-pro')
|
| 22 |
|
| 23 |
+
client = Client("https://alanchen1115-gemini-api.hf.space/--replicas/mgfss/", hf_token=os.environ[HF_TOKEN], serialize=False)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
|
| 25 |
app = FastAPI()
|
| 26 |
app.add_middleware(
|