thangquang09 commited on
Commit ·
09a8622
1
Parent(s): c23bdf9
update Dockerfile
Browse files- Dockerfile +1 -1
- app.py +1 -1
Dockerfile
CHANGED
|
@@ -8,7 +8,7 @@ RUN pip install --no-cache-dir --upgrade -r /requirements.txt
|
|
| 8 |
|
| 9 |
RUN curl -fsSL https://ollama.com/install.sh | sh
|
| 10 |
|
| 11 |
-
RUN ollama pull
|
| 12 |
|
| 13 |
COPY start.sh /start.sh
|
| 14 |
RUN chmod +x /start.sh
|
|
|
|
| 8 |
|
| 9 |
RUN curl -fsSL https://ollama.com/install.sh | sh
|
| 10 |
|
| 11 |
+
RUN ollama pull phi2
|
| 12 |
|
| 13 |
COPY start.sh /start.sh
|
| 14 |
RUN chmod +x /start.sh
|
app.py
CHANGED
|
@@ -12,7 +12,7 @@ class Generate(BaseModel):
|
|
| 12 |
duration: float
|
| 13 |
|
| 14 |
chat_history = []
|
| 15 |
-
model = Ollama(model="
|
| 16 |
|
| 17 |
def generate_text(model: Ollama, prompt: str) -> {}:
|
| 18 |
if prompt == "":
|
|
|
|
| 12 |
duration: float
|
| 13 |
|
| 14 |
chat_history = []
|
| 15 |
+
model = Ollama(model="phi2")
|
| 16 |
|
| 17 |
def generate_text(model: Ollama, prompt: str) -> {}:
|
| 18 |
if prompt == "":
|