mertbozkurt commited on
Commit
305bee5
·
1 Parent(s): ce17f5f
Files changed (2) hide show
  1. app.py +31 -0
  2. requirements.txt +4 -0
app.py ADDED
@@ -0,0 +1,31 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import AutoTokenizer
3
+ import transformers
4
+ import torch
5
+
6
+ prompt_input = st.text_input("Prompt")
7
+
8
+
9
+ model = "mertbozkurt/Llama-2-7b-TR-recipe"
10
+ prompt = "Sodalı köfte nasıl yapımı?"
11
+
12
+ tokenizer = AutoTokenizer.from_pretrained(model,offload_folder="offload")
13
+ pipeline = transformers.pipeline(
14
+ "text-generation",
15
+ model=model,
16
+ torch_dtype=torch.float16,
17
+ device_map="auto",
18
+ offload_folder="offload"
19
+ )
20
+
21
+ sequences = pipeline(
22
+ f'<s>[INST] {prompt} [/INST]',
23
+ do_sample=True,
24
+ top_k=10,
25
+ num_return_sequences=1,
26
+ eos_token_id=tokenizer.eos_token_id,
27
+ max_length=100,
28
+ )
29
+ for seq in sequences:
30
+ st.write(seq)
31
+
requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ transformers
2
+ accelerate
3
+ torch
4
+ streamlit