Update app.py

app.py

@@ -25,6 +25,7 @@ agents =[
 ]
 def generate(prompt, history):
     print(f'HISTORY:: {history}')
+    history=[]
     output1={}
     seed = random.randint(1,1111111111111111)
 
@@ -46,26 +47,7 @@ def generate(prompt, history):
     for response in stream:
         output += response.token.text
         yield [(prompt,output)],output1
-    
-    '''
-    with open('tmp.txt','w') as t:
-        t.writelines(output)
-    t.close()
-    with open('tmp.txt','r') as t:
-        out_lines = t.readlines()
-    t.close()
-    
-    for i,line in enumerate(out_lines):
-        print(line)
-        if "DIFFUSION_PROMPT" in line:
-            try:
-                d_prompt = line.split("DIFFUSION_PROMPT:",1)[1].split("MEME_TEXT",1)[0]
-                print(d_prompt)
-            except Exception as e:
-                print(e)
-                pass
-    '''
-    output1=output.replace("\n"," ")
+  
     if "PROMPT:" and "MEME_TEXT:" in output1:
         print("YES")
         prompt_t=output1.split("PROMPT:",1)[1].split("MEME_TEXT:",1)[0].strip()
@@ -89,42 +71,6 @@ def run(inp,model_drop):
         out = Image.open(io.BytesIO(r.content))
     return out
 
-def run_OG(inp,model_drop):
-    prompt=inp['PROMPT']
-    #if len(inp)>max_prompt:
-    #    inp = condense(inp)
-    #print(f'history :: {history}')
-    #prompt=format_prompt(inp,history)
-    seed = random.randint(1,1111111111111111)
-    print (seed)
-    generate_kwargs = dict(
-        temperature=1.0,
-        max_new_tokens=1048,
-        top_p=0.99,
-        repetition_penalty=1.0,
-        do_sample=True,
-        seed=seed,
-    )
-    content = GENERATE_PROMPT + prompt
-    print(content)
-    stream = client.text_generation(content, **generate_kwargs, stream=True, details=True, return_full_text=False)
-    resp = ""
-    for response in stream:
-        resp += response.token.text
-
-        yield ([(inp,resp)],None)
-    model=loaded_model[int(model_drop)]
-    out_img=model(resp)
-    print(out_img)
-    url=f'https://omnibus-meme-diffusion.hf.space/file={out_img}'
-    print(url)
-    uid = uuid.uuid4()
-    r = requests.get(url, stream=True)
-    if r.status_code == 200:
-        out = Image.open(io.BytesIO(r.content))
-    yield ([(inp,resp)],out)
-
-
 def run_gpt(in_prompt,history,):
     if len(in_prompt)>max_prompt:
         in_prompt = condense(in_prompt)