Spaces:
Running
Running
- app.py +2 -0
- app_cosmos.py +2 -0
app.py
CHANGED
@@ -4,6 +4,8 @@ import torch
|
|
4 |
from huggingface_hub import login
|
5 |
import os
|
6 |
|
|
|
|
|
7 |
# read access token from environment variable
|
8 |
access_token = os.getenv('HF_TOKEN')
|
9 |
login(access_token)
|
|
|
4 |
from huggingface_hub import login
|
5 |
import os
|
6 |
|
7 |
+
print("Google Gemma 2 Chatbot is starting...")
|
8 |
+
|
9 |
# read access token from environment variable
|
10 |
access_token = os.getenv('HF_TOKEN')
|
11 |
login(access_token)
|
app_cosmos.py
CHANGED
@@ -2,6 +2,8 @@ from fastapi import FastAPI, Request
|
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
import torch
|
4 |
|
|
|
|
|
5 |
model_id = "ytu-ce-cosmos/Turkish-Llama-8b-DPO-v0.1"
|
6 |
|
7 |
print("Model loading started")
|
|
|
2 |
from transformers import AutoTokenizer, AutoModelForCausalLM
|
3 |
import torch
|
4 |
|
5 |
+
print("COSMOS Llama Chatbot is starting...")
|
6 |
+
|
7 |
model_id = "ytu-ce-cosmos/Turkish-Llama-8b-DPO-v0.1"
|
8 |
|
9 |
print("Model loading started")
|